neuralmagic / sparsezoo

Neural network model repository for highly sparse and sparse-quantized models with matching sparsification recipes
Apache License 2.0
371 stars 25 forks source link

[BugFix] Add analyze to init #421

Closed rahul-tuli closed 10 months ago

rahul-tuli commented 10 months ago

Before Analyze wasn't importable from sparsezoo; now it is

from sparsezoo import analyze
analyze("zoo:codegen_mono-350m-bigpython_bigquery_thepile-pruned50_quantized")

This revealed another bug in analyze, we are taking a look at that

Needs the following as well to get the complete bugfix: