Skip to content

Commit 8d30d25

Browse files
authored
[LoRA] default to None when fc alphas are not available. (huggingface#4706)
default to None when fc alphas are not available.
1 parent 1e0395e commit 8d30d25

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/diffusers/loaders.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1400,8 +1400,9 @@ def _modify_text_encoder(
14001400

14011401
if patch_mlp:
14021402
for name, mlp_module in text_encoder_mlp_modules(text_encoder):
1403-
fc1_alpha = network_alphas.pop(name + ".fc1.lora_linear_layer.down.weight.alpha")
1404-
fc2_alpha = network_alphas.pop(name + ".fc2.lora_linear_layer.down.weight.alpha")
1403+
fc1_alpha = network_alphas.pop(name + ".fc1.lora_linear_layer.down.weight.alpha", None)
1404+
fc2_alpha = network_alphas.pop(name + ".fc2.lora_linear_layer.down.weight.alpha", None)
1405+
14051406
current_rank_fc1 = rank.pop(f"{name}.fc1.lora_linear_layer.up.weight")
14061407
current_rank_fc2 = rank.pop(f"{name}.fc2.lora_linear_layer.up.weight")
14071408

0 commit comments

Comments
 (0)