remove deprecated message (#1619)

pytorch · Sep 17, 2024 · 0dae85f · 0dae85f
1 parent 7cf9b26
commit 0dae85f
Showing 1 changed file with 0 additions and 6 deletions.
diff --git a/torchtune/modules/transformer.py b/torchtune/modules/transformer.py
@@ -11,7 +11,6 @@
 from torch import nn
 from torchtune.modules import MultiHeadAttention
 from torchtune.modules.attention_utils import _MaskType
-from torchtune.utils.logging import deprecated
 
 
 class TransformerSelfAttentionLayer(nn.Module):
@@ -515,11 +514,6 @@ def forward(
         return output
 
 
-@deprecated(
-    msg="Please use torchtune.modules.TransformerDecoder instead. \
-If you need an example, see torchtune.models.qwen2._component_builders.py \
-and how to implement torch.modules.TiedLinear for the output projection."
-)
 class TiedEmbeddingTransformerDecoder(nn.Module):
     """
     Transformer Decoder with tied embedding weight. A key difference between