allow conversion from no state dict checkpoints

d67c3051 · Patrick von Platen · 2bd53a94 · d67c3051
Commit d67c3051 authored Jan 03, 2023 by Patrick von Platen
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

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

No files found.
--- a/scripts/convert_original_stable_diffusion_to_diffusers.py
+++ b/scripts/convert_original_stable_diffusion_to_diffusers.py
@@ -866,7 +866,9 @@ if __name__ == "__main__":
    else:
        print("global_step key not found in model")
        global_step = None
-    checkpoint = checkpoint["state_dict"]
+
+    if "state_dict" in checkpoint:
+        checkpoint = checkpoint["state_dict"]

    upcast_attention = False
    if args.original_config_file is None: