We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d811296 commit 39c56e3Copy full SHA for 39c56e3
src/deepsparse/transformers/helpers.py
@@ -74,6 +74,9 @@ def get_deployment_path(model_path: str) -> Tuple[str, str]:
74
elif model_path.startswith("zoo:"):
75
zoo_model = Model(model_path)
76
deployment_path = zoo_model.deployment.path
77
+ for deployment_file in zoo_model.deployment.files:
78
+ # force download of any missing files in deployment directory
79
+ deployment_file.path
80
return deployment_path, os.path.join(deployment_path, _MODEL_DIR_ONNX_NAME)
81
elif model_path.startswith("hf:"):
82
from huggingface_hub import snapshot_download
0 commit comments