Skip to content

Commit 54ef06a

Browse files
committed
fix layer sequential pipeline
Signed-off-by: Kyle Sayers <[email protected]>
1 parent 58fe929 commit 54ef06a

File tree

1 file changed

+2
-2
lines changed
  • src/llmcompressor/pipelines/layer_sequential

1 file changed

+2
-2
lines changed

src/llmcompressor/pipelines/layer_sequential/pipeline.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
to_next_layer_kwargs,
1717
)
1818
from llmcompressor.pipelines.registry import CalibrationPipeline
19-
from llmcompressor.pipelines.sequential.helpers import get_targets_from_modifiers
19+
from llmcompressor.pipelines.sequential.helpers import get_sequential_targets
2020
from llmcompressor.utils.helpers import DisableQuantization, calibration_forward_context
2121

2222
if TYPE_CHECKING:
@@ -68,7 +68,7 @@ def __call__(
6868

6969
# find layers
7070
modifiers = session.get_modifiers()
71-
sequential_targets, _ = get_targets_from_modifiers(modifiers, model)
71+
sequential_targets = get_sequential_targets(modifiers, model, dataset_args)
7272
layers = match_modules(model, sequential_targets)
7373

7474
LifecycleCallbacks.calibration_epoch_start()

0 commit comments

Comments
 (0)