Formatting

This commit is contained in:
Nicolas Mowen 2024-09-17 14:42:04 -06:00
parent bb305c6d9d
commit 4b12e03ab1

View File

@ -54,7 +54,9 @@ class Clip(OnnxClip):
for provider in providers: for provider in providers:
if provider == "TensorrtExecutionProvider": if provider == "TensorrtExecutionProvider":
os.makedirs("/config/model_cache/tensorrt/ort/trt-engines", exist_ok=True) os.makedirs(
"/config/model_cache/tensorrt/ort/trt-engines", exist_ok=True
)
options.append( options.append(
{ {
"trt_timing_cache_enable": True, "trt_timing_cache_enable": True,