neuralmagic / deepsparse

Sparsity-aware deep learning inference runtime for CPUs
https://neuralmagic.com/deepsparse/
Other
3.01k stars 176 forks source link

[bugfix] enforce that all transformer deployement files are downloaded by pipeline #1303

Closed bfineran closed 1 year ago

bfineran commented 1 year ago

recent changes simplify downloading transformers deployment files by just running zoo_model.depoyment.path while this works in the base case, @dbarbuzzi found that if some files are already downloaded (ie just onnx files for benchmarking) the additional configs will not be downloaded by this command.

this pr simply loops over the deployment directory's contents to makes sure that all files get downloaded before any pipelines run

test_plan: manually verified