[PEFT / LoRA] Fix civitai bug when network alpha is an empty dict (…

…huggingface#5608) * fix civitai bug * add test * up * fix test * added slow test. * style * Update src/diffusers/utils/peft_utils.py Co-authored-by: Benjamin Bossan <[email protected]> * Update src/diffusers/utils/peft_utils.py --------- Co-authored-by: Benjamin Bossan <[email protected]>
yoonseokjin · Nov 1, 2023 · 1be7e62 · 1be7e62
1 parent ffef55d
commit 1be7e62
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/peft_utils.py b/utils/peft_utils.py
@@ -129,7 +129,7 @@ def get_peft_kwargs(rank_dict, network_alpha_dict, peft_state_dict, is_unet=True
         rank_pattern = dict(filter(lambda x: x[1] != r, rank_dict.items()))
         rank_pattern = {k.split(".lora_B.")[0]: v for k, v in rank_pattern.items()}
 
-    if network_alpha_dict is not None:
+    if network_alpha_dict is not None and len(network_alpha_dict) > 0:
         if len(set(network_alpha_dict.values())) > 1:
             # get the alpha occuring the most number of times
             lora_alpha = collections.Counter(network_alpha_dict.values()).most_common()[0][0]