File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
src/llmcompressor/pipelines/layer_sequential Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 16
16
to_next_layer_kwargs ,
17
17
)
18
18
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
20
20
from llmcompressor .utils .helpers import DisableQuantization , calibration_forward_context
21
21
22
22
if TYPE_CHECKING :
@@ -68,7 +68,7 @@ def __call__(
68
68
69
69
# find layers
70
70
modifiers = session .get_modifiers ()
71
- sequential_targets , _ = get_targets_from_modifiers (modifiers , model )
71
+ sequential_targets = get_sequential_targets (modifiers , model , dataset_args )
72
72
layers = match_modules (model , sequential_targets )
73
73
74
74
LifecycleCallbacks .calibration_epoch_start ()
You can’t perform that action at this time.
0 commit comments