Update README.md

README.md

@@ -12,6 +12,6 @@ Below is the description of the model folders:
 - `mp_causal_60000.pt`: Multiplayer causal model weights produced by training Stage 3
 - `mp_bidirectional_120000.pt`: Multiplayer bidirectional model weights produced by training Stage 2
 - `sp_bidirectional_pretrain_120000.pt`: Single player bidirectional model weights produced by training Stage 1
-- `matrix-game-init`: [Matrix Game 2.0 weights](https://huggingface.co/Skywork/Matrix-Game-2.0/tree/main/base_distilled_model) with an extended action space. Used for single player pretraining (Stage 1) initialization
+- `matrix-game-init`: [Matrix Game 2.0 weights](https://huggingface.co/Skywork/Matrix-Game-2.0/tree/main/base_distilled_model) with the keyboard action MLP dimension increased to 23 (randmoly initialized). Used for single player pretraining (Stage 1) initialization
 
 See [Solaris](https://github.com/solaris-wm/solaris?tab=readme-ov-file#training-stages) repo for a detailed description of the training stages.