Skip to content

Commit

Permalink
remove sparsezoo
Browse files Browse the repository at this point in the history
  • Loading branch information
horheynm committed Apr 12, 2024
1 parent b195681 commit 0458417
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ def _setup_install_requires() -> List:
return ["torch>=1.7.0", "transformers<=4.40", "pydantic<2.7"]

def _setup_extras() -> Dict:
return {"dev": ["black==22.12.0", "isort==5.8.0", "wheel>=0.36.2", "flake8>=3.8.3", "pytest>=6.0.0", "sparsezoo==1.7.0"]}
return {"dev": ["black==22.12.0", "isort==5.8.0", "wheel>=0.36.2", "flake8>=3.8.3", "pytest>=6.0.0",]}

setup(
name="sparsetensors",
Expand Down
2 changes: 1 addition & 1 deletion src/sparsetensors/quantization/observers/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
from typing import Optional, Tuple

from sparsetensors.quantization.quant_args import QuantizationArgs
from sparsezoo.utils.registry import RegistryMixin
from sparsetensors.registry.registry import RegistryMixin
from torch import FloatTensor, IntTensor, Tensor
from torch.nn import Module

Expand Down

0 comments on commit 0458417

Please sign in to comment.