Skip to content

GitLab

Unverified Commit 3ce6380d authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

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

parent d2dc4de3
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
scripts/convert_original_stable_diffusion_to_diffusers.py
View file @ 3ce6380d
...@@ -925,6 +925,8 @@ if __name__ == "__main__": ...@@ -925,6 +925,8 @@ if __name__ == "__main__":
set_alpha_to_one=False, set_alpha_to_one=False,
prediction_type=prediction_type, prediction_type=prediction_type,
) )
# make sure scheduler works correctly with DDIM
scheduler.register_to_config(clip_sample=False)
if args.scheduler_type == "pndm": if args.scheduler_type == "pndm":
config = dict(scheduler.config) config = dict(scheduler.config)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment