support fluxgym loras

nunchaku/lora/flux/comfyui_converter.py

@@ -19,7 +19,7 @@ def comfyui2diffusers(
     new_tensors = {}
     max_rank = 0
     for k, v in tensors.items():
-        if "alpha" in k:
+        if "alpha" in k or "lora_te" in k:
             continue
         new_k = k.replace("lora_down", "lora_A").replace("lora_up", "lora_B")
         if "lora_unet_double_blocks_" in k:

nunchaku/lora/flux/utils.py

@@ -12,7 +12,7 @@ def detect_format(lora: str | dict[str, torch.Tensor]) -> str:
     for k in tensors.keys():
         if "lora_unet_double_blocks_" in k or "lora_unet_single_blocks" in k:
             return "comfyui"
-        elif "mlp_fc" in k or "mlp_context_fc1" in k:
+        elif ".mlp_fc" in k or "mlp_context_fc1" in k:
             return "svdquant"
         elif "double_blocks." in k or "single_blocks." in k:
             return "xlab"