Fix convert_original_stable_diffusion_to_diffusers script (#4817)

Fix stable diffusion conversion script

Fix convert_original_stable_diffusion_to_diffusers script (#4817)
Fix stable diffusion conversion script
3eeaf4e0 · Alexsey Shestacov · GitHub · c583f3b4 · 3eeaf4e0
Unverified Commit 3eeaf4e0 authored Aug 29, 2023 by Alexsey Shestacov Committed by GitHub Aug 29, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

scripts/convert_original_stable_diffusion_to_diffusers.py scripts/convert_original_stable_diffusion_to_diffusers.py +1 -1

No files found.
--- a/scripts/convert_original_stable_diffusion_to_diffusers.py
+++ b/scripts/convert_original_stable_diffusion_to_diffusers.py
@@ -152,7 +152,7 @@ if __name__ == "__main__":
        pipeline_class = None
    pipe = download_from_original_stable_diffusion_ckpt(
-        checkpoint_path=args.checkpoint_path,
+        checkpoint_path_or_dict=args.checkpoint_path,
        original_config_file=args.original_config_file,
        image_size=args.image_size,
        prediction_type=args.prediction_type,