Formatting

This commit is contained in:
Nicolas Mowen 2024-10-29 07:09:38 -06:00
parent ea7f9854a7
commit b2ed2fea4f

View File

@ -47,7 +47,8 @@ def get_ort_providers(
options.append(
{
"arena_extend_strategy": "kSameAsRequested",
"trt_fp16_enable": requires_fp16 and os.environ.get("USE_FP_16", "True") != "False",
"trt_fp16_enable": requires_fp16
and os.environ.get("USE_FP_16", "True") != "False",
"trt_timing_cache_enable": True,
"trt_engine_cache_enable": True,
"trt_timing_cache_path": "/config/model_cache/tensorrt/ort",