ok

server.py

@@ -476,11 +476,14 @@ class GameManager:
             
             # Create a basic inner model config
             inner_model_cfg = InnerModelConfig(
-                hidden_channels=128,
                 img_channels=3,
                 num_steps_conditioning=4,
-                is_upsampler=False,
-                num_actions=6  # Default number of actions
+                cond_channels=128,
+                depths=[2, 2, 2],
+                channels=[64, 128, 256],
+                attn_depths=[False, False, True],
+                num_actions=6,  # Default number of actions
+                is_upsampler=False
             )
             
             # Create denoiser config and instance