diff --git a/integration-tests/models/test_bloom_560m.py b/integration-tests/models/test_bloom_560m.py index 3c598c04..809250cb 100644 --- a/integration-tests/models/test_bloom_560m.py +++ b/integration-tests/models/test_bloom_560m.py @@ -9,7 +9,7 @@ def bloom_560_handle(launcher): @pytest.fixture(scope="module") async def bloom_560(bloom_560_handle): - await bloom_560_handle.health(60) + await bloom_560_handle.health(240) return bloom_560_handle.client diff --git a/integration-tests/models/test_bloom_560m_sharded.py b/integration-tests/models/test_bloom_560m_sharded.py index 25f6b2d7..ee67250a 100644 --- a/integration-tests/models/test_bloom_560m_sharded.py +++ b/integration-tests/models/test_bloom_560m_sharded.py @@ -9,7 +9,7 @@ def bloom_560m_sharded_handle(launcher): @pytest.fixture(scope="module") async def bloom_560m_sharded(bloom_560m_sharded_handle): - await bloom_560m_sharded_handle.health(60) + await bloom_560m_sharded_handle.health(240) return bloom_560m_sharded_handle.client diff --git a/integration-tests/models/test_flash_falcon.py b/integration-tests/models/test_flash_falcon.py index ce27731d..e36a6a28 100644 --- a/integration-tests/models/test_flash_falcon.py +++ b/integration-tests/models/test_flash_falcon.py @@ -9,7 +9,7 @@ def flash_falcon_handle(launcher): @pytest.fixture(scope="module") async def flash_falcon(flash_falcon_handle): - await flash_falcon_handle.health(120) + await flash_falcon_handle.health(300) return flash_falcon_handle.client diff --git a/integration-tests/models/test_flash_llama.py b/integration-tests/models/test_flash_llama.py index bf5b64ba..edc847c1 100644 --- a/integration-tests/models/test_flash_llama.py +++ b/integration-tests/models/test_flash_llama.py @@ -9,7 +9,7 @@ def flash_llama_handle(launcher): @pytest.fixture(scope="module") async def flash_llama(flash_llama_handle): - await flash_llama_handle.health(120) + await flash_llama_handle.health(300) return flash_llama_handle.client diff --git a/integration-tests/models/test_flash_neox.py b/integration-tests/models/test_flash_neox.py index ff9915ed..daff7f0a 100644 --- a/integration-tests/models/test_flash_neox.py +++ b/integration-tests/models/test_flash_neox.py @@ -9,7 +9,7 @@ def flash_neox_handle(launcher): @pytest.fixture(scope="module") async def flash_neox(flash_neox_handle): - await flash_neox_handle.health(240) + await flash_neox_handle.health(300) return flash_neox_handle.client diff --git a/integration-tests/models/test_flash_neox_sharded.py b/integration-tests/models/test_flash_neox_sharded.py index 513aeaaf..a1aa0f07 100644 --- a/integration-tests/models/test_flash_neox_sharded.py +++ b/integration-tests/models/test_flash_neox_sharded.py @@ -9,7 +9,7 @@ def flash_neox_sharded_handle(launcher): @pytest.fixture(scope="module") async def flash_neox_sharded(flash_neox_sharded_handle): - await flash_neox_sharded_handle.health(240) + await flash_neox_sharded_handle.health(300) return flash_neox_sharded_handle.client diff --git a/integration-tests/models/test_flash_santacoder.py b/integration-tests/models/test_flash_santacoder.py index b0cb4522..a15a6439 100644 --- a/integration-tests/models/test_flash_santacoder.py +++ b/integration-tests/models/test_flash_santacoder.py @@ -9,7 +9,7 @@ def flash_santacoder_handle(launcher): @pytest.fixture(scope="module") async def flash_santacoder(flash_santacoder_handle): - await flash_santacoder_handle.health(240) + await flash_santacoder_handle.health(300) return flash_santacoder_handle.client diff --git a/integration-tests/models/test_flash_starcoder.py b/integration-tests/models/test_flash_starcoder.py index c1a68d89..72b298c9 100644 --- a/integration-tests/models/test_flash_starcoder.py +++ b/integration-tests/models/test_flash_starcoder.py @@ -9,7 +9,7 @@ def flash_starcoder_handle(launcher): @pytest.fixture(scope="module") async def flash_starcoder(flash_starcoder_handle): - await flash_starcoder_handle.health(240) + await flash_starcoder_handle.health(300) return flash_starcoder_handle.client diff --git a/integration-tests/models/test_mt0_base.py b/integration-tests/models/test_mt0_base.py index e347d22a..4ed95aad 100644 --- a/integration-tests/models/test_mt0_base.py +++ b/integration-tests/models/test_mt0_base.py @@ -9,7 +9,7 @@ def mt0_base_handle(launcher): @pytest.fixture(scope="module") async def mt0_base(mt0_base_handle): - await mt0_base_handle.health(60) + await mt0_base_handle.health(300) return mt0_base_handle.client diff --git a/integration-tests/models/test_t5_sharded.py b/integration-tests/models/test_t5_sharded.py index 074660c7..a2d84330 100644 --- a/integration-tests/models/test_t5_sharded.py +++ b/integration-tests/models/test_t5_sharded.py @@ -9,7 +9,7 @@ def t5_sharded_handle(launcher): @pytest.fixture(scope="module") async def t5_sharded(t5_sharded_handle): - await t5_sharded_handle.health(240) + await t5_sharded_handle.health(300) return t5_sharded_handle.client