Fix megablox tile sizes

Author: gobbleturk

tests/unit/train_compile_test.py

@@ -973,4 +973,4 @@ def test_qk_clip(self):
             "use_qk_clip=true",
             "qk_clip_threshold=100",
         )
-    )
+    )