fix load keys in forge

2025-01-22 19:34:11 +09:00 · 2025-01-22 19:34:11 +09:00 · 5a0f745ed6
parent e946d3c1b3
commit 5a0f745ed6
1 changed files with 1 additions and 1 deletions
--- a/scripts/supermerger.py
+++ b/scripts/supermerger.py
@ -1063,7 +1063,7 @@ def modeltype(sd):
 def loadkeys(model_a, lora):
    if lora:
        import lora
-        sd = sd_models.read_state_dict(lora.available_loras[model_a].filename,"cpu")
+        sd =  load_torch_file(lora.available_loras[model_a].filename, device = torch.device("cpu")) if forge else sd_models.read_state_dict(lora.available_loras[model_a].filename,"cpu")
    else:
        sd = loadmodel(model_a)
    keys = []