[SD] Make sure scheduler is correct when converting (open-mmlab#1667)

aravind-h-v · Dec 12, 2022 · 3ce6380 · 3ce6380
1 parent d2dc4de
commit 3ce6380
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/scripts/convert_original_stable_diffusion_to_diffusers.py b/scripts/convert_original_stable_diffusion_to_diffusers.py
@@ -925,6 +925,8 @@ def convert_open_clip_checkpoint(checkpoint):
         set_alpha_to_one=False,
         prediction_type=prediction_type,
     )
+    # make sure scheduler works correctly with DDIM
+    scheduler.register_to_config(clip_sample=False)
 
     if args.scheduler_type == "pndm":
         config = dict(scheduler.config)