update token positions

pi_zero_pytorch/pi_zero.py

@@ -53,7 +53,7 @@
 # vision and language tokens are autoregressive causal mask, actions, interal states + joint bidirectional amongst own tokens, but still autoregressive with respect to other tokens
 
 # [state token groups] [action token groups]
-# [external state] [visual tokens] [language tokens] [joint state + internal state] [maybe reward / condition token] [action registers] [actions]
+# [external state] [visual tokens] [language tokens] [maybe reward / condition token] [action registers] [joint state + internal state] [actions]
 
 # constants