LoHA: Fixed blending error

ssube · Apr 8, 2023 · d244ca9 · d244ca9
1 parent bd03edc
commit d244ca9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api/onnx_web/convert/diffusion/lora.py b/api/onnx_web/convert/diffusion/lora.py
@@ -267,8 +267,8 @@ def blend_loras(
             else:
                 if base_weights.shape != weights.shape:
                     blended = base_weights + weights.reshape(base_weights.shape)
-
-                blended = base_weights + weights
+                else:
+                    blended = base_weights + weights
 
             logger.trace("blended weight shape: %s", blended.shape)