Skip to content

Commit d0e5bc5

Browse files
remove hardcoded pipeline logic
Signed-off-by: Brian Dellabetta <bdellabe@redhat.com>
1 parent f2ef7cf commit d0e5bc5

File tree

1 file changed

+0
-5
lines changed

1 file changed

+0
-5
lines changed

src/llmcompressor/pipelines/registry.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88

99
from llmcompressor.modifiers import Modifier
1010
from llmcompressor.modifiers.quantization import QuantizationModifier
11-
from llmcompressor.modifiers.transform import SpinQuantModifier
1211

1312
if TYPE_CHECKING:
1413
from llmcompressor.args.dataset_arguments import DatasetArguments
@@ -62,8 +61,4 @@ def _infer_pipeline(modifiers: List[Modifier]) -> str:
6261
if not config.requires_calibration_data():
6362
return "datafree"
6463

65-
# TODO: Remove hardcode
66-
if len(modifiers) == 1 and isinstance(modifiers[0], SpinQuantModifier):
67-
return "datafree"
68-
6964
return "sequential"

0 commit comments

Comments
 (0)