Fix unmerged lora

oops
2025-09-02 18:48:25 +03:00
parent 3183589eff
commit c538cd4c8a
1 changed files with 1 additions and 1 deletions
@@ -107,7 +107,7 @@ class CustomLinear(nn.Linear):
            ).reshape(weight.shape)
            alpha = lora_diff[2] / lora_diff[1].shape[0] if lora_diff[2] is not None else 1.0
            scale = lora_strength * alpha
-            weight.add(patch_diff, alpha=scale)
+            weight = weight.add(patch_diff, alpha=scale)
        return weight
    
 def remove_lora_from_module(module):