theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-23 00:58:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

update flux lora convert state dict

Browse Source
This commit is contained in:
lzw478614@alibaba-inc.com
2025-07-14 14:08:22 +08:00
parent 44f6ffbf56
commit dc066aca2d
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
diffsynth/lora/flux_lora.py
View File

@@ -102,12 +102,14 @@ class FluxLoRALoader(GeneralLoRALoader):
for k in state_dict: for k in state_dict:
if "lora_unet_" in k: if "lora_unet_" in k:
return 'civitai' return 'civitai'
elif "transformer." in k: elif k.startswith("transformer."):
return 'diffusers' return 'diffusers'
else: else:
None None
model_resource = guess_resource(state_dict) model_resource = guess_resource(state_dict)
if model_resource is None:
return state_dict
rename_dict = self.diffusers_rename_dict if model_resource == 'diffusers' else self.civitai_rename_dict rename_dict = self.diffusers_rename_dict if model_resource == 'diffusers' else self.civitai_rename_dict
def guess_alpha(state_dict): def guess_alpha(state_dict):