From ae5e40785eaaa5ca32c53466eab6ccf410544c9c Mon Sep 17 00:00:00 2001 From: Seunghoon Lee Date: Mon, 30 Oct 2023 18:49:11 +0900 Subject: [PATCH] fix ep option save --- modules/shared.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/shared.py b/modules/shared.py index 13987f22a..6f30a5b80 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -438,7 +438,7 @@ options_templates.update(options_section(('diffusers', "Diffusers Settings"), { "huggingface_token": OptionInfo('', 'HuggingFace token'), "onnx_sep": OptionInfo("

ONNX Runtime

", "", gr.HTML), - "onnx_execution_provider": OptionInfo(get_default_execution_provider(), 'Execution Provider', gr.Dropdown, lambda: {"choices": available_execution_providers }), + "onnx_execution_provider": OptionInfo(get_default_execution_provider().value, 'Execution Provider', gr.Dropdown, lambda: {"choices": available_execution_providers }), "onnx_olive_float16": OptionInfo(True, 'Use FP16 on Olive optimization (will use FP32 if unchecked)'), "onnx_cache_optimized": OptionInfo(True, 'Cache Olive optimized models'), }))