diff --git a/integration-tests/models/test_flash_gemma3.py b/integration-tests/models/test_flash_gemma3.py index 022d2c47..ab812d64 100644 --- a/integration-tests/models/test_flash_gemma3.py +++ b/integration-tests/models/test_flash_gemma3.py @@ -3,7 +3,7 @@ import pytest @pytest.fixture(scope="module") def flash_gemma3_handle(launcher): - with launcher("gg-hf-g/gemma-3-4b-it", num_shard=2) as handle: + with launcher("google/gemma-3-4b-it", num_shard=2) as handle: yield handle