Skip to content

Commit

Permalink
download compressed deployment directory (#1318)
Browse files Browse the repository at this point in the history
  • Loading branch information
bfineran authored Oct 17, 2023
1 parent 1f97dc4 commit 697752f
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 9 deletions.
5 changes: 1 addition & 4 deletions src/deepsparse/transformers/helpers.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,10 +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
for deployment_file in zoo_model.deployment.files:
# force download of any missing files in deployment directory
deployment_file.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
Expand Down
7 changes: 2 additions & 5 deletions src/deepsparse/utils/onnx.py
Original file line number Diff line number Diff line change
Expand Up @@ -127,11 +127,8 @@ 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
# trigger download and unzipping of deployment directory if not cached
model.deployment_directory_path

# default to the main onnx file for the model
model = model.deployment.get_file(_MODEL_DIR_ONNX_NAME).path
Expand Down

0 comments on commit 697752f

Please sign in to comment.