apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix variable name

This commit is contained in:
Sean Sube 2023-09-24 18:05:48 -05:00
parent e1c2ae5b1b
commit cdb09d2b44
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/diffusers/load.py
View File

@ -424,7 +424,7 @@ def load_unet(server, device, model, loras, unet_type, params):
logger.info("blending base model %s with LoRA models: %s", model, lora_models)
# blend and load unet
blended_unet = blend_loras(
unet = blend_loras(
server,
unet,
list(zip(lora_models, lora_weights)),
@ -432,7 +432,7 @@ def load_unet(server, device, model, loras, unet_type, params):
xl=params.is_xl(),
)
(unet_model, unet_data) = buffer_external_data_tensors(blended_unet)
(unet_model, unet_data) = buffer_external_data_tensors(unet)
unet_names, unet_values = zip(*unet_data)
unet_opts = device.sess_options(cache=False)
unet_opts.add_external_initializers(list(unet_names), list(unet_values))