"Fixing t5" just use more RAM for this test.

2024-04-12 08:46:33 +00:00 · 2024-04-12 08:46:33 +00:00 · 1e5150f475
parent cd07211411
commit 1e5150f475
1 changed files with 1 additions and 1 deletions
--- a/integration-tests/models/test_t5_sharded.py
+++ b/integration-tests/models/test_t5_sharded.py
@ -3,7 +3,7 @@ import pytest

@pytest.fixture(scope="module")
 def t5_sharded_handle(launcher):
-    with launcher("google/flan-t5-xxl", num_shard=2) as handle:
+    with launcher("google/flan-t5-xxl", num_shard=4) as handle:
        yield handle