Fix noise scheduler

2022-11-15 11:08:38 -05:00 · 2022-11-15 11:08:38 -05:00 · 29ffbd645e
parent 5be5a487b2
commit 29ffbd645e
1 changed files with 1 additions and 0 deletions
--- a/trainer/diffusers_trainer.py
+++ b/trainer/diffusers_trainer.py
@ -690,6 +690,7 @@ def main():
        beta_end=0.012,
        beta_schedule='scaled_linear',
        num_train_timesteps=1000,
+        clip_sample=False
    )

    # load dataset