up

engine.py

@@ -50,8 +50,15 @@ class MatrixGameEngine:
             args: Optional parsed command line arguments for model configuration
         """
         # Set default parameters if args not provided
-        self.frame_width = getattr(args, 'frame_width', 640)
-        self.frame_height = getattr(args, 'frame_height', 360)
+        # Ensure frame dimensions are compatible with VAE downsampling (8x) and patch size [1,2,2]
+        # Dimensions must be divisible by vae_scale_factor * patch_size = 8 * 2 = 16
+        default_width = getattr(args, 'frame_width', 640)
+        default_height = getattr(args, 'frame_height', 368)  # Changed from 360 to 368 (368/16=23)
+        
+        # Ensure compatibility with VAE and patch size
+        vae_patch_factor = 16  # vae_scale_factor (8) * patch_size (2) for both H and W
+        self.frame_width = (default_width // vae_patch_factor) * vae_patch_factor
+        self.frame_height = (default_height // vae_patch_factor) * vae_patch_factor
         self.fps = getattr(args, 'fps', 16)
         self.inference_steps = getattr(args, 'inference_steps', 20)
         self.guidance_scale = getattr(args, 'guidance_scale', 6.0)