Small fix

BenjaminBossan · BenjaminBossan · commit 8bebb061124f · 2025-06-30T16:16:52.000+02:00
diff --git a/src/diffusers/loaders/unet_loader_utils.py b/src/diffusers/loaders/unet_loader_utils.py
@@ -156,8 +156,9 @@ def _maybe_expand_lora_scales_for_one_adapter(
 
         del scales[updown]
 
+    state_dict = model.state_dict()
     for layer in scales.keys():
-        if not any(_translate_into_actual_layer_name(layer) in module for module in model.state_dict().keys()):
+        if not any(_translate_into_actual_layer_name(layer) in module for module in state_dict.keys()):
             raise ValueError(
                 f"Can't set lora scale for layer {layer}. It either doesn't exist in this unet or it has no attentions."
             )

Original file line number	Diff line number	Diff line change
`@@ -156,8 +156,9 @@ def _maybe_expand_lora_scales_for_one_adapter(`
`156`	`156`
`157`	`157`	`del scales[updown]`
`158`	`158`
	`159`	`+ state_dict = model.state_dict()`
`159`	`160`	`for layer in scales.keys():`
`160`		`- if not any(_translate_into_actual_layer_name(layer) in module for module in model.state_dict().keys()):`
	`161`	`+ if not any(_translate_into_actual_layer_name(layer) in module for module in state_dict.keys()):`
`161`	`162`	`raise ValueError(`
`162`	`163`	`f"Can't set lora scale for layer {layer}. It either doesn't exist in this unet or it has no attentions."`
`163`	`164`	`)`