Update helpers.py (#1351)

neuralmagic · Oct 25, 2023 · fd5dbb5 · fd5dbb5
1 parent 42bffa3
commit fd5dbb5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/deepsparse/transformers/helpers.py b/src/deepsparse/transformers/helpers.py
@@ -73,7 +73,7 @@ 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
+        deployment_path = zoo_model.deployment_directory_path
         return deployment_path, os.path.join(deployment_path, _MODEL_DIR_ONNX_NAME)
     elif model_path.startswith("hf:"):
         from huggingface_hub import snapshot_download