search

neuralmagic / sparseml

Libraries for applying sparsification recipes to neural networks with a few lines of code, enabling faster and smaller models
Apache License 2.0
2.05k stars 144 forks source link

Fix import typo #2311

Closed eldarkurtic closed 4 months ago

eldarkurtic commented 4 months ago

from sparseml import SparseAutoModelForCausalLM errors with:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ImportError: cannot import name 'SparseAutoModelForCausalLM' from 'sparseml' (/home/eldar/to_delete/sparseml/src/sparseml/__init__.py)

I think it should be from sparseml.transformers import SparseAutoModelForCausalLM