We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 89818be commit 959a173Copy full SHA for 959a173
modelopt/torch/_deploy/_runtime/tensorrt/engine_builder.py
@@ -213,8 +213,11 @@ def _setup_files_and_paths(
213
final_output_dir = Path(output_dir or Path(gettempdir()) / DEFAULT_ARTIFACT_DIR)
214
final_output_dir.mkdir(parents=True, exist_ok=True)
215
engine_path = (
216
- engine_path if engine_path else final_output_dir / f"{onnx_bytes.model_name}.engine"
+ Path(engine_path)
217
+ if engine_path
218
+ else final_output_dir / f"{onnx_bytes.model_name}.engine"
219
)
220
+ engine_path.parent.mkdir(parents=True, exist_ok=True)
221
calib_cache_path = final_output_dir / "calib_cache" if calib_cache else None
222
timing_cache_path = final_output_dir / "timing.cache"
223
0 commit comments