fix test_model

Lightning-AI · Jan 4, 2025 · 1ac888f · 1ac888f
1 parent 276a8fc
commit 1ac888f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_model.py b/tests/test_model.py
@@ -645,6 +645,7 @@ def test_against_olmo2(model_name, device, dtype):
         n_layer=2,
         n_head=8,
         n_embd=32,
+        n_query_groups=2,
         intermediate_size=86,
     )
     T = 5