From 18e25798421ea0e6094739cdbebba82f837465e7 Mon Sep 17 00:00:00 2001 From: Seunghoon Lee Date: Mon, 30 Oct 2023 18:18:09 +0900 Subject: [PATCH] update footprint filename of cpuep --- modules/onnx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/onnx.py b/modules/onnx.py index 7501aeaaf..203906db3 100644 --- a/modules/onnx.py +++ b/modules/onnx.py @@ -25,7 +25,7 @@ submodels = ("text_encoder", "unet", "vae_encoder", "vae_decoder",) available_execution_providers: List[ExecutionProvider] = ort.get_available_providers() EP_TO_NAME = { - ExecutionProvider.CPU: "cpu?", # TODO + ExecutionProvider.CPU: "cpu", ExecutionProvider.DirectML: "gpu-dml", ExecutionProvider.CUDA: "gpu-?", # TODO ExecutionProvider.ROCm: "gpu-rocm",