fix: use spatial_merge_size = 1 in unit test

Author: imkero

tests/models/qwen2_5_omni/test_modeling_qwen2_5_omni.py

@@ -77,7 +77,7 @@ def __init__(
             "mlp_ratio": 4,
             "num_heads": 4,
             "patch_size": 14,
-            "spatial_merge_size": 2,
+            "spatial_merge_size": 1,
             "temporal_patch_size": 2,
             "fullatt_block_indexes": [0],
             "initializer_range": 0.02,
@@ -385,7 +385,7 @@ def test_get_rope_index_video_with_audio(self):
         image_grid_thw = torch.empty((0, 3), dtype=torch.long)
 
         # 3 * 2 * 2 = 12 video tokens
-        video_grid_thw = torch.tensor([[3, 4, 4]], dtype=torch.long)
+        video_grid_thw = torch.tensor([[3, 2, 2]], dtype=torch.long)
 
         # num_audio_tokens = ((audio_seqlen - 1) // 2 + 1 - 2) // 2 + 1
         # i.e.: 300 audio_seqlen -> 75 audio tokens