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

Fix noise scheduler

This commit is contained in:
Anthony Mercurio 2022-11-15 11:08:38 -05:00 committed by GitHub
parent 5be5a487b2
commit 29ffbd645e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
trainer/diffusers_trainer.py
View File

@ -690,6 +690,7 @@ 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,
clip_sample=False
) )
# load dataset # load dataset