From cdb09d2b444f6665fd2127776a8a74240b2f41ed Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 24 Sep 2023 18:05:48 -0500 Subject: [PATCH] fix variable name --- api/onnx_web/diffusers/load.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/diffusers/load.py b/api/onnx_web/diffusers/load.py index f51b191c..37800e71 100644 --- a/api/onnx_web/diffusers/load.py +++ b/api/onnx_web/diffusers/load.py @@ -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))