Skip to content

Commit 91b84b0

Browse files
committed
Removing unused decoder for TransformedModel class.
1 parent a287899 commit 91b84b0

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

model_pytorch.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -157,8 +157,6 @@ def __init__(self, cfg, vocab=40990, n_ctx=512):
157157
self.drop = nn.Dropout(cfg.embd_pdrop)
158158
block = Block(n_ctx, cfg, scale=True)
159159
self.h = nn.ModuleList([copy.deepcopy(block) for _ in range(cfg.n_layer)])
160-
self.decoder = nn.Linear(cfg.n_embd, vocab, bias=False)
161-
self.decoder.weight = self.embed.weight # Tied weights
162160

163161
nn.init.normal_(self.embed.weight, std=0.02)
164162

0 commit comments

Comments
 (0)