Skip to content

Commit 8351ac9

Browse files
committed
name change
Signed-off-by: Kyle Sayers <[email protected]>
1 parent 6fdcdb1 commit 8351ac9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/llmcompressor/pipelines/registry.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
__all__ = ["CalibrationPipeline"]
1919

2020
SEQUENTIAL_MODIFIERS = (AWQModifier, GPTQModifier, SparsityModifierBase)
21-
NEED_DATA = (SmoothQuantModifier, *SEQUENTIAL_MODIFIERS)
21+
CALIBRATION_MODIFIERS = (SmoothQuantModifier, *SEQUENTIAL_MODIFIERS)
2222

2323

2424
class CalibrationPipeline(ABC, RegistryMixin):
@@ -61,7 +61,7 @@ def from_modifiers(
6161

6262
@staticmethod
6363
def _validate_infer_pipeline(modifiers: List[Modifier]) -> str:
64-
if any(isinstance(modifier, NEED_DATA) for modifier in modifiers):
64+
if any(isinstance(modifier, CALIBRATION_MODIFIERS) for modifier in modifiers):
6565
return "sequential"
6666

6767
active_qmods = _get_active_quant_modifiers(modifiers)

0 commit comments

Comments
 (0)