apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

add context arg

This commit is contained in:
Sean Sube 2023-02-25 12:13:22 -06:00
parent d88f13cbd7
commit 973b0a9352
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -207,5 +207,6 @@ def merge_lora():
if __name__ == "__main__": if __name__ == "__main__":
convert_diffusion_lora("unet") context = ConversionContext.from_environ()
convert_diffusion_lora("text_encoder") convert_diffusion_lora(context, "unet")
convert_diffusion_lora(context, "text_encoder")