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

Fix issue where LoRa loads on a device different from Dit

Browse Source
This commit is contained in:
yaozhengbing
2025-12-31 21:31:01 +08:00
parent ab8580f77e
commit 60ac96525b
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
diffsynth/utils/lora/flux.py
View File

@@ -149,6 +149,8 @@ class FluxLoRALoader(GeneralLoRALoader):
dtype=state_dict_[name].dtype) dtype=state_dict_[name].dtype)
else: else:
state_dict_.pop(name.replace(".a_to_q.", ".proj_in_besides_attn.")) state_dict_.pop(name.replace(".a_to_q.", ".proj_in_besides_attn."))
mlp = mlp.to(device=state_dict_[name].device)
if 'lora_A' in name: if 'lora_A' in name:
param = torch.concat([ param = torch.concat([
state_dict_.pop(name), state_dict_.pop(name),