[bugfix] enforce that all transformer deployement files are downloade…

…d by pipeline (#1303)
neuralmagic · Oct 10, 2023 · 39c56e3 · 39c56e3
1 parent d811296
commit 39c56e3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/deepsparse/transformers/helpers.py b/src/deepsparse/transformers/helpers.py
@@ -74,6 +74,9 @@ def get_deployment_path(model_path: str) -> Tuple[str, str]:
     elif model_path.startswith("zoo:"):
         zoo_model = Model(model_path)
         deployment_path = zoo_model.deployment.path
+        for deployment_file in zoo_model.deployment.files:
+            # force download of any missing files in deployment directory
+            deployment_file.path
         return deployment_path, os.path.join(deployment_path, _MODEL_DIR_ONNX_NAME)
     elif model_path.startswith("hf:"):
         from huggingface_hub import snapshot_download