From 1e5150f475df4a069eb2e7c0be4ddb4a2c872973 Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Fri, 12 Apr 2024 08:46:33 +0000 Subject: [PATCH] "Fixing t5" just use more RAM for this test. --- integration-tests/models/test_t5_sharded.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/integration-tests/models/test_t5_sharded.py b/integration-tests/models/test_t5_sharded.py index 7c288b23..4b4cfd98 100644 --- 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