Fix issue where LoRa loads on a device different from Dit

2026-03-18 22:08:13 +00:00 · 2025-12-31 21:31:01 +08:00
parent ab8580f77e
commit 60ac96525b
1 changed files with 2 additions and 0 deletions
--- a/diffsynth/utils/lora/flux.py
+++ b/diffsynth/utils/lora/flux.py
@@ -149,6 +149,8 @@ class FluxLoRALoader(GeneralLoRALoader):
                                        dtype=state_dict_[name].dtype)
                    else:
                        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:
                        param = torch.concat([
                            state_dict_.pop(name),