Skip to content

Commit f8c15dc

Browse files
committed
fix typo
Signed-off-by: Kyle Sayers <[email protected]>
1 parent e3e009f commit f8c15dc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/llmcompressor/utils/helpers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1103,7 +1103,7 @@ def requires_lm_head_calibration(
11031103
if isinstance(mod, QuantizationMixin):
11041104
targets |= set(match_named_modules(model, mod.resolved_targets, mod.ignore))
11051105

1106-
return targets_embeddings(model, targets, check_input=True, check_output=False)
1106+
return targets_embeddings(model, targets, check_input=False, check_output=True)
11071107

11081108

11091109
@contextlib.contextmanager

0 commit comments

Comments
 (0)