up

Author: sayakpaul

src/diffusers/quantizers/auto.py

@@ -21,7 +21,7 @@
 
 from .bitsandbytes import BnB4BitDiffusersQuantizer, BnB8BitDiffusersQuantizer
 from .gguf import GGUFQuantizer
-from .nunchaku import NunChakuQuantizer
+from .nunchaku import NunchakuQuantizer
 from .quantization_config import (
     BitsAndBytesConfig,
     GGUFQuantizationConfig,
@@ -41,7 +41,7 @@
     "gguf": GGUFQuantizer,
     "quanto": QuantoQuantizer,
     "torchao": TorchAoHfQuantizer,
-    "nunchaku": NunChakuQuantizer,
+    "nunchaku": NunchakuQuantizer,
 }
 
 AUTO_QUANTIZATION_CONFIG_MAPPING = {

src/diffusers/quantizers/nunchaku/__init__.py

@@ -1 +1 @@
-from .nunchaku_quantizer import NunChakuQuantizer
+from .nunchaku_quantizer import NunchakuQuantizer

src/diffusers/quantizers/nunchaku/nunchaku_quantizer.py

@@ -29,7 +29,7 @@
 logger = logging.get_logger(__name__)
 
 
-class NunChakuQuantizer(DiffusersQuantizer):
+class NunchakuQuantizer(DiffusersQuantizer):
     r"""
     Diffusers Quantizer for Nunchaku (https://github.com/nunchaku-tech/nunchaku)
     """