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")