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

Merge pull request #685 from lzws/main

Browse Source 
update flux lora convert state dict
This commit is contained in:
Zhongjie Duan
2025-07-14 14:58:07 +08:00
committed by GitHub
parent 44f6ffbf56 dc066aca2d
commit 836fa5c957
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:
if "lora_unet_" in k:
return 'civitai'
elif "transformer." in k:
elif k.startswith("transformer."):
return 'diffusers'
else:
None
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
def guess_alpha(state_dict):