rename piecewise to sequential

vllm-project · Dec 9, 2024 · f8591ca · f8591ca
1 parent e468197
commit f8591ca
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 3 deletions.
diff --git a/src/llmcompressor/modifiers/quantization/gptq/base.py b/src/llmcompressor/modifiers/quantization/gptq/base.py
@@ -26,7 +26,7 @@
 from llmcompressor.modifiers.quantization.quantization.base import QuantizationModifier
 from llmcompressor.modifiers.utils.hooks import HooksMixin
 from llmcompressor.pipelines.basic import run_pipeline as run_basic
-from llmcompressor.pipelines.piecewise import run_pipeline as run_piecewise
+from llmcompressor.pipelines.sequential import run_pipeline as run_sequential
 from llmcompressor.utils.metric_logging import CompressionLogger
 from llmcompressor.utils.pytorch.module import (
     get_layers,
@@ -222,7 +222,7 @@ def on_initialize(self, state: "State", **kwargs) -> bool:
         # infer pipeline
         if "pixel_values" not in state.data.calib.dataset.column_names:
             try:
-                run_piecewise(
+                run_sequential(
                     state.model,
                     self.sequential_targets,
                     self.ignore,

diff --git a/...ompressor/pipelines/piecewise/__init__.py → ...mpressor/pipelines/sequential/__init__.py b/...ompressor/pipelines/piecewise/__init__.py → ...mpressor/pipelines/sequential/__init__.py
diff --git a/...compressor/pipelines/piecewise/helpers.py → ...ompressor/pipelines/sequential/helpers.py b/...compressor/pipelines/piecewise/helpers.py → ...ompressor/pipelines/sequential/helpers.py
diff --git a/...ompressor/pipelines/piecewise/pipeline.py → ...mpressor/pipelines/sequential/pipeline.py b/...ompressor/pipelines/piecewise/pipeline.py → ...mpressor/pipelines/sequential/pipeline.py
@@ -7,7 +7,7 @@
 
 from llmcompressor.modifiers.utils.hooks import HooksMixin
 from llmcompressor.modifiers.utils.pytorch_helpers import apply_pad_mask_to_batch
-from llmcompressor.pipelines.piecewise.helpers import (
+from llmcompressor.pipelines.sequential.helpers import (
     infer_sequential_targets,
     trace_subgraphs,
 )