Skip to content

Commit b46655b

Browse files
authored
[Misc] Remove deprecation warning (#2107)
## Purpose ## * Remove deprecation decorator which was accidentally added to `disable_lm_head` Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
1 parent 4ecc2af commit b46655b

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

src/llmcompressor/utils/helpers.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1093,7 +1093,6 @@ def calibration_forward_context(model: torch.nn.Module):
10931093
yield
10941094

10951095

1096-
@deprecated()
10971096
@contextlib.contextmanager
10981097
def disable_lm_head(model: torch.nn.Module):
10991098
"""

0 commit comments

Comments
 (0)