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 f2ef7cf commit d0e5bc5Copy full SHA for d0e5bc5
src/llmcompressor/pipelines/registry.py
@@ -8,7 +8,6 @@
8
9
from llmcompressor.modifiers import Modifier
10
from llmcompressor.modifiers.quantization import QuantizationModifier
11
-from llmcompressor.modifiers.transform import SpinQuantModifier
12
13
if TYPE_CHECKING:
14
from llmcompressor.args.dataset_arguments import DatasetArguments
@@ -62,8 +61,4 @@ def _infer_pipeline(modifiers: List[Modifier]) -> str:
62
61
if not config.requires_calibration_data():
63
return "datafree"
64
65
- # TODO: Remove hardcode
66
- if len(modifiers) == 1 and isinstance(modifiers[0], SpinQuantModifier):
67
- return "datafree"
68
-
69
return "sequential"
0 commit comments