From 973b0a93524a2f7fd6a4031b54110349cff6a284 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 25 Feb 2023 12:13:22 -0600 Subject: [PATCH] add context arg --- api/onnx_web/convert/diffusion/lora.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/convert/diffusion/lora.py b/api/onnx_web/convert/diffusion/lora.py index f0144f3d..57e85d79 100644 --- a/api/onnx_web/convert/diffusion/lora.py +++ b/api/onnx_web/convert/diffusion/lora.py @@ -207,5 +207,6 @@ def merge_lora(): if __name__ == "__main__": - convert_diffusion_lora("unet") - convert_diffusion_lora("text_encoder") + context = ConversionContext.from_environ() + convert_diffusion_lora(context, "unet") + convert_diffusion_lora(context, "text_encoder")