We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6fdcdb1 commit 8351ac9Copy full SHA for 8351ac9
src/llmcompressor/pipelines/registry.py
@@ -18,7 +18,7 @@
18
__all__ = ["CalibrationPipeline"]
19
20
SEQUENTIAL_MODIFIERS = (AWQModifier, GPTQModifier, SparsityModifierBase)
21
-NEED_DATA = (SmoothQuantModifier, *SEQUENTIAL_MODIFIERS)
+CALIBRATION_MODIFIERS = (SmoothQuantModifier, *SEQUENTIAL_MODIFIERS)
22
23
24
class CalibrationPipeline(ABC, RegistryMixin):
@@ -61,7 +61,7 @@ def from_modifiers(
61
62
@staticmethod
63
def _validate_infer_pipeline(modifiers: List[Modifier]) -> str:
64
- if any(isinstance(modifier, NEED_DATA) for modifier in modifiers):
+ if any(isinstance(modifier, CALIBRATION_MODIFIERS) for modifier in modifiers):
65
return "sequential"
66
67
active_qmods = _get_active_quant_modifiers(modifiers)
0 commit comments