From 3ce6380d3a2ec5c3e3f4f48889d380d657b151bc Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Mon, 12 Dec 2022 16:57:48 +0100 Subject: [PATCH] [SD] Make sure scheduler is correct when converting (#1667) --- scripts/convert_original_stable_diffusion_to_diffusers.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/scripts/convert_original_stable_diffusion_to_diffusers.py b/scripts/convert_original_stable_diffusion_to_diffusers.py index ebb3327d1c..b027262029 100644 --- a/scripts/convert_original_stable_diffusion_to_diffusers.py +++ b/scripts/convert_original_stable_diffusion_to_diffusers.py @@ -925,6 +925,8 @@ if __name__ == "__main__": 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)