From e669abac85840f9f61b54e2c5326ab9f320af2d1 Mon Sep 17 00:00:00 2001 From: xadupre Date: Sun, 8 Jun 2025 22:30:56 +0200 Subject: [PATCH] Fix precision name for modelbuilder --- onnx_diagnostic/helpers/model_builder_helper.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/onnx_diagnostic/helpers/model_builder_helper.py b/onnx_diagnostic/helpers/model_builder_helper.py index 29839998..c87c28e1 100644 --- a/onnx_diagnostic/helpers/model_builder_helper.py +++ b/onnx_diagnostic/helpers/model_builder_helper.py @@ -220,6 +220,9 @@ def create_model_builder( """ assert cache_dir, "create_model_builder does not work without cache_dir." assert os.path.exists(cache_dir), f"cache_dir={cache_dir!r} does not exists" + precision = {"float32": "fp32", "float16": "fp16", "bfloat16": "bfp16"}.get( + precision, precision + ) download_model_builder_to_cache() builder = import_model_builder() io_dtype = builder.set_io_dtype(precision, execution_provider, extra_options)