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