[SD] Make sure scheduler is correct when converting (huggingface#1667)

patrickvonplaten · web-flow · commit 3ce6380d3a2e · 2022-12-12T16:57:48.000+01:00
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)

Original file line number	Diff line number	Diff line change
`@@ -925,6 +925,8 @@ def convert_open_clip_checkpoint(checkpoint):`
`925`	`925`	`set_alpha_to_one=False,`
`926`	`926`	`prediction_type=prediction_type,`
`927`	`927`	`)`
	`928`	`+ # make sure scheduler works correctly with DDIM`
	`929`	`+ scheduler.register_to_config(clip_sample=False)`
`928`	`930`
`929`	`931`	`if args.scheduler_type == "pndm":`
`930`	`932`	`config = dict(scheduler.config)`