fix: prefer meta-llama/Llama-2-7b-hf over deprecated repo

This commit is contained in:
drbh
2025-11-06 19:42:45 +00:00
parent ae1fb28434
commit 27bc1271d1
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -3,7 +3,7 @@ import pytest
@pytest.fixture(scope="module")
def flash_llama_handle(launcher):
with launcher("huggingface/llama-7b", num_shard=2) as handle:
with launcher("meta-llama/Llama-2-7b-hf", num_shard=2) as handle:
yield handle
+1 -1
View File
@@ -14,7 +14,7 @@ def get_test_model():
def generate_token(self, batch):
raise NotImplementedError
tokenizer = AutoTokenizer.from_pretrained("huggingface/llama-7b")
tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-hf")
model = TestModel(
"test_model_id",