apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

apply lint

This commit is contained in:
Sean Sube 2023-04-20 21:26:16 -05:00
parent ff4c8a9404
commit 49781df2fd
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
api/onnx_web/convert/diffusion/lora.py
View File

@ -173,7 +173,11 @@ def blend_loras(
np_weights *= lora_weight
if base_key in blended:
logger.trace("summing LoHA weights: %s + %s", blended[base_key].shape, np_weights.shape)
logger.trace(
"summing LoHA weights: %s + %s",
blended[base_key].shape,
np_weights.shape,
)
blended[base_key] += sum_weights(blended[base_key], np_weights)
else:
blended[base_key] = np_weights
@ -282,7 +286,11 @@ def blend_loras(
np_weights *= lora_weight
if base_key in blended:
logger.trace("summing weights: %s + %s", blended[base_key].shape, np_weights.shape)
logger.trace(
"summing weights: %s + %s",
blended[base_key].shape,
np_weights.shape,
)
blended[base_key] = sum_weights(blended[base_key], np_weights)
else:
blended[base_key] = np_weights