removed extra blocks call in goal-conditioned gpt

jannerm · Feb 15, 2022 · c79460b · c79460b
1 parent e0b5f12
commit c79460b
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/trajectory/models/transformers.py b/trajectory/models/transformers.py
@@ -290,7 +290,6 @@ def forward(self, idx, goal, targets=None, mask=None):
         x = gx[:, self.observation_dim:]
         #### /goal
 
-        x = self.blocks(x)
         ## [ B x T x embedding_dim ]
         x = self.ln_f(x)