[Cherry Pick] Download deployment dir for llms (#436)

rahul-tuli · web-flow · commit fccb742a9d16 · 2024-02-09T13:50:29.000-05:00
* Download deployment dir for llms

* Use path instead of download
diff --git a/src/sparsezoo/analyze/analysis.py b/src/sparsezoo/analyze/analysis.py
@@ -1305,6 +1305,8 @@ def create(cls, file_path: Union[str, ModelProto]) -> "ModelAnalysis":
             result = ModelAnalysis.from_onnx(Path(file_path) / "model.onnx")
 
         elif file_path.startswith("zoo:"):
+            # download and extract deployment directory
+            Model(file_path).deployment.path
             result = ModelAnalysis.from_onnx(
                 Model(file_path).deployment.get_file("model.onnx").path
             )

Original file line number	Diff line number	Diff line change
`@@ -1305,6 +1305,8 @@ def create(cls, file_path: Union[str, ModelProto]) -> "ModelAnalysis":`
`1305`	`1305`	`result = ModelAnalysis.from_onnx(Path(file_path) / "model.onnx")`
`1306`	`1306`
`1307`	`1307`	`elif file_path.startswith("zoo:"):`
	`1308`	`+ # download and extract deployment directory`
	`1309`	`+ Model(file_path).deployment.path`
`1308`	`1310`	`result = ModelAnalysis.from_onnx(`
`1309`	`1311`	`Model(file_path).deployment.get_file("model.onnx").path`
`1310`	`1312`	`)`