docker

huggingface · Feb 5, 2024 · e5ad8e3 · e5ad8e3
1 parent ae5c3bd
commit e5ad8e3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/models/llama/test_inference_llama.py b/tests/models/llama/test_inference_llama.py
@@ -11,7 +11,7 @@
 class LLaMaInferenceTest(unittest.TestCase):
 
     def test_foo(self):
-        ckpt = "meta-llama/Llama-2-7b"
+        ckpt = "meta-llama/Llama-2-7b-hf"
         assert os.getenv("HF_HUB_READ_TOKEN", None) != None
         assert type(os.getenv("HF_HUB_READ_TOKEN", None)) == str
         assert len(os.getenv("HF_HUB_READ_TOKEN", None)) > 3

diff --git a/tests/models/llama/test_train_llama.py b/tests/models/llama/test_train_llama.py
@@ -9,7 +9,7 @@
 class LLaMaTrainingTest(unittest.TestCase):
 
     def test_foo(self):
-        ckpt = "meta-llama/Llama-2-7b"
+        ckpt = "meta-llama/Llama-2-7b-hf"
         assert os.getenv("HF_HUB_READ_TOKEN", None) != None
         assert type(os.getenv("HF_HUB_READ_TOKEN", None)) == str
         assert len(os.getenv("HF_HUB_READ_TOKEN", None)) > 3