From f82de3339e11ad7d8db4e0ec20716bbcdaa5af74 Mon Sep 17 00:00:00 2001
From: sayakpaul <spsayakpaul@gmail.com>
Date: Tue, 24 Jun 2025 16:25:20 +0530
Subject: [PATCH] updates

---
 src/diffusers/quantizers/auto.py                  | 3 ++-
 src/diffusers/quantizers/higgs/__init__.py        | 1 +
 src/diffusers/quantizers/higgs/higgs_quantizer.py | 2 +-
 src/diffusers/quantizers/quantization_config.py   | 1 +
 4 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/diffusers/quantizers/auto.py b/src/diffusers/quantizers/auto.py
index d04e4a98a7..9a69690dfb 100644
--- a/src/diffusers/quantizers/auto.py
+++ b/src/diffusers/quantizers/auto.py
@@ -21,6 +21,7 @@ from typing import Dict, Optional, Union
 
 from .bitsandbytes import BnB4BitDiffusersQuantizer, BnB8BitDiffusersQuantizer
 from .gguf import GGUFQuantizer
+from .higgs import HiggsQuantizer
 from .quantization_config import (
     BitsAndBytesConfig,
     GGUFQuantizationConfig,
@@ -40,7 +41,7 @@ AUTO_QUANTIZER_MAPPING = {
     "gguf": GGUFQuantizer,
     "quanto": QuantoQuantizer,
     "torchao": TorchAoHfQuantizer,
-    "higgs": 1,
+    "higgs": HiggsQuantizer,
 }
 
 AUTO_QUANTIZATION_CONFIG_MAPPING = {
diff --git a/src/diffusers/quantizers/higgs/__init__.py b/src/diffusers/quantizers/higgs/__init__.py
index e69de29bb2..37dbbf0141 100644
--- a/src/diffusers/quantizers/higgs/__init__.py
+++ b/src/diffusers/quantizers/higgs/__init__.py
@@ -0,0 +1 @@
+from .higgs_quantizer import HiggsQuantizer
diff --git a/src/diffusers/quantizers/higgs/higgs_quantizer.py b/src/diffusers/quantizers/higgs/higgs_quantizer.py
index 8261ad5af7..8a1705857d 100644
--- a/src/diffusers/quantizers/higgs/higgs_quantizer.py
+++ b/src/diffusers/quantizers/higgs/higgs_quantizer.py
@@ -35,7 +35,7 @@ if is_torch_available():
 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.
diff --git a/src/diffusers/quantizers/quantization_config.py b/src/diffusers/quantizers/quantization_config.py
index c6d3740339..6518a9eae3 100644
--- a/src/diffusers/quantizers/quantization_config.py
+++ b/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():