From 8276b8995ab14442258be0163dad74cada67c24a Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Wed, 5 Jun 2024 16:23:18 +0200 Subject: [PATCH] Only 1 GPU for now. --- integration-tests/models/test_flash_gpt2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/integration-tests/models/test_flash_gpt2.py b/integration-tests/models/test_flash_gpt2.py index 0c7977d0..1e906521 100644 --- a/integration-tests/models/test_flash_gpt2.py +++ b/integration-tests/models/test_flash_gpt2.py @@ -3,7 +3,7 @@ import pytest @pytest.fixture(scope="module") def flash_gpt2_handle(launcher): - with launcher("openai-community/gpt2", num_shard=2) as handle: + with launcher("openai-community/gpt2", num_shard=1) as handle: yield handle