Skip to content

Commit eafc28a

Browse files
authored
Merge pull request #37 from rodgzilla/removing_unused_decoder
Removing unused decoder for TransformedModel class.
2 parents a287899 + 91b84b0 commit eafc28a

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)