Mirrors
/
waifu-diffusion
mirror of https://github.com/harubaru/waifu-diffusion.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

diffusers no longer use tensor_format in DDPMScheduler

This commit is contained in:
Astralite Heart 2022-11-05 11:05:45 +00:00
parent 74ef98667c
commit 0e0acf31b1
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
diffusers_trainer.py
View File

@ -523,7 +523,6 @@ def main():
beta_end=0.012, beta_end=0.012,
beta_schedule='scaled_linear', beta_schedule='scaled_linear',
num_train_timesteps=1000, num_train_timesteps=1000,
tensor_format='pt'
) )
# load dataset # load dataset