Merge pull request #1171 from YZBPXX/main

Artiprocher · web-flow · commit a6884f6b3aa0 · 2026-01-05T16:39:02.000+08:00
Fix issue where LoRa loads on a device different from Dit
diff --git a/diffsynth/utils/lora/flux.py b/diffsynth/utils/lora/flux.py
@@ -149,6 +149,8 @@ def guess_alpha(state_dict):
                                         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),
