onnx fix

lshqqytiger · Jun 5, 2024 · 73a4e8c · 73a4e8c
1 parent 2c29feb
commit 73a4e8c
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 0 deletions.
diff --git a/models/ONNX/.gitkeep b/models/ONNX/.gitkeep
diff --git a/modules/shared_options.py b/modules/shared_options.py
@@ -214,6 +214,7 @@
 options_templates.update(options_section(('onnx', "ONNX Runtime", "sd"), {
     "onnx_enable": OptionInfo(False, 'Use ONNX Runtime instead of PyTorch implementation'),
     "diffusers_pipeline": OptionInfo('ONNX Stable Diffusion', 'Diffusers pipeline', gr.Dropdown, lambda: {"choices": list(shared_items.get_pipelines())}),
+    "diffusers_vae_upcast": OptionInfo("default", "VAE upcasting", gr.Radio, {"choices": ['default', 'true', 'false']}),
     "onnx_execution_provider": OptionInfo(get_default_execution_provider().value, 'Execution Provider', gr.Dropdown, lambda: {"choices": available_execution_providers }),
     "onnx_cache_converted": OptionInfo(True, 'ONNX cache converted models'),