diff --git a/api/onnx_web/convert/diffusion/lora.py b/api/onnx_web/convert/diffusion/lora.py index ae1503d03..d2b842556 100644 --- 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)