[bugfix] model_to_path should also download .data files from zoo (#1296)

neuralmagic · Oct 4, 2023 · 09ad1d9 · 09ad1d9
1 parent 5d022a7
commit 09ad1d9
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/deepsparse/utils/onnx.py b/src/deepsparse/utils/onnx.py
@@ -127,6 +127,12 @@ def model_to_path(model: Union[str, Model, File]) -> str:
         model = Model(model)
 
     if Model is not object and isinstance(model, Model):
+        # download any onnx data files in deployment directory
+        for deployment_file in model.deployment.files:
+            if ".data" in deployment_file.name:
+                # forces download of data file if not cached
+                deployment_file.path
+
         # default to the main onnx file for the model
         model = model.deployment.get_file(_MODEL_DIR_ONNX_NAME).path