Skip to content

Commit 70da261

Browse files
committed
reduce diff
Signed-off-by: Kyle Sayers <[email protected]>
1 parent 6a13bc4 commit 70da261

File tree

1 file changed

+0
-16
lines changed
  • src/compressed_tensors/quantization/utils

1 file changed

+0
-16
lines changed

src/compressed_tensors/quantization/utils/helpers.py

Lines changed: 0 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,6 @@
3333

3434

3535
__all__ = [
36-
"infer_quantization_status",
3736
"is_module_quantized",
3837
"is_model_quantized",
3938
"module_type",
@@ -236,21 +235,6 @@ def calculate_range(quantization_args: QuantizationArgs, device: str) -> Tuple:
236235
return q_min, q_max
237236

238237

239-
def infer_quantization_status(model: Module) -> Optional["QuantizationStatus"]: # noqa
240-
"""
241-
Checks the quantization status of a model. Assumes all modules in the model have
242-
the same status, so only the first quantized model is checked.
243-
244-
:param model: model to check quantization status for
245-
:return: quantization status if the model is quantized, otherwise None
246-
"""
247-
for module in model.modules():
248-
status = getattr(module, "quantization_status", None)
249-
if status is not None:
250-
return status
251-
return None
252-
253-
254238
def is_module_quantized(module: Module) -> bool:
255239
"""
256240
Check if a module is quantized, based on the existence of a non-empty quantization

0 commit comments

Comments
 (0)