ONNX/Olive support .safetensors VAE.

This commit is contained in:
Seunghoon Lee
2024-01-14 17:23:02 +09:00
parent 0fac69cbf5
commit ea2ac32cf2
6 changed files with 282 additions and 230 deletions
+1
View File
@@ -444,6 +444,7 @@ options_templates.update(options_section(('diffusers', "Diffusers Settings"), {
"onnx_sep": OptionInfo("<h2>ONNX Runtime</h2>", "", gr.HTML),
"onnx_execution_provider": OptionInfo(get_default_execution_provider().value, 'Execution Provider', gr.Dropdown, lambda: {"choices": available_execution_providers }),
"onnx_show_menu": OptionInfo(False, 'ONNX show onnx-specific menu (restart required)'),
"onnx_cache_converted": OptionInfo(True, 'ONNX cache converted models'),
"onnx_unload_base": OptionInfo(False, 'ONNX unload base model when processing refiner'),
}))