updates

Author: sayakpaul

src/diffusers/quantizers/auto.py

@@ -21,6 +21,7 @@
 
 from .bitsandbytes import BnB4BitDiffusersQuantizer, BnB8BitDiffusersQuantizer
 from .gguf import GGUFQuantizer
+from .higgs import HiggsQuantizer
 from .quantization_config import (
     BitsAndBytesConfig,
     GGUFQuantizationConfig,
@@ -40,7 +41,7 @@
     "gguf": GGUFQuantizer,
     "quanto": QuantoQuantizer,
     "torchao": TorchAoHfQuantizer,
-    "higgs": 1,
+    "higgs": HiggsQuantizer,
 }
 
 AUTO_QUANTIZATION_CONFIG_MAPPING = {

src/diffusers/quantizers/higgs/__init__.py

@@ -0,0 +1 @@
+from .higgs_quantizer import HiggsQuantizer

src/diffusers/quantizers/higgs/higgs_quantizer.py

@@ -35,7 +35,7 @@
 logger = logging.get_logger(__name__)
 
 
-class HiggsHfQuantizer(DiffusersQuantizer):
+class HiggsQuantizer(DiffusersQuantizer):
     """
     Quantizer of the HIGGS method. Enables the loading of prequantized models and in-flight quantization of
     full-precision models.

src/diffusers/quantizers/quantization_config.py

@@ -46,6 +46,7 @@ class QuantizationMethod(str, Enum):
     GGUF = "gguf"
     TORCHAO = "torchao"
     QUANTO = "quanto"
+    HIGGS = "higgs"
 
 
 if is_torchao_available():