fix sd model convert bug (#644)

mindspore-lab · Aug 29, 2024 · 89256b1 · 89256b1
1 parent bc82f54
commit 89256b1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/stable_diffusion_v2/tools/model_conversion/convert_weights.py b/examples/stable_diffusion_v2/tools/model_conversion/convert_weights.py
@@ -52,7 +52,7 @@
 
 def _load_torch_ckpt(ckpt_file):
     source_data = torch.load(ckpt_file, map_location="cpu")
-    if ["state_dict"] in source_data:
+    if "state_dict" in source_data:
         source_data = source_data["state_dict"]
     return source_data