From 29ffbd645e0b441a1d7f15085e39fefc7ecbce99 Mon Sep 17 00:00:00 2001
From: Anthony Mercurio <seizou.genkei@protonmail.com>
Date: Tue, 15 Nov 2022 11:08:38 -0500
Subject: [PATCH] Fix noise scheduler

---
 trainer/diffusers_trainer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/trainer/diffusers_trainer.py b/trainer/diffusers_trainer.py
index 9a6e2b7..c0194b9 100644
--- 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