From 84748a01e6d3c26037412592ce147b7753ce6117 Mon Sep 17 00:00:00 2001
From: Francois Fleuret <francois@fleuret.org>
Date: Tue, 26 Jul 2022 12:47:26 +0200
Subject: [PATCH] Removed the Linear transformation since there is now w_o.

---
 mygpt.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/mygpt.py b/mygpt.py
index 7f0c9e6..5370ffa 100755
--- a/mygpt.py
+++ b/mygpt.py
@@ -111,7 +111,6 @@ class MyGPT(nn.Module):
                         nb_heads = nb_heads,
                         causal = True, attention_dropout = dropout
                     ),
-                    nn.Linear(in_features = dim_model, out_features = dim_model),
                 ),
                 Residual(
                     nn.LayerNorm(dim_model),
-- 
2.39.5