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 0eba539 commit c26ca8bCopy full SHA for c26ca8b
python/torch_mlir/tools/import_onnx/__main__.py
@@ -137,7 +137,7 @@ def load_onnx_model(args: argparse.Namespace) -> onnx.ModelProto:
137
# Load the temp file and the external data.
138
inferred_model = onnx.load(temp_inferred_file, load_external_data=False)
139
data_dir = Path(input_dir if args.temp_dir is None else args.data_dir)
140
- onnx.load_external_data_for_model(inferred_model, data_dir)
+ onnx.load_external_data_for_model(inferred_model, str(data_dir))
141
142
# Remove the inferred shape file unless asked to keep it
143
if not args.keep_temps:
0 commit comments