Use unrestricted tokenizer

argilla-io · Aug 6, 2024 · f317128 · f317128
1 parent 3f40170
commit f317128
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/unit/llms/test_vllm.py b/tests/unit/llms/test_vllm.py
@@ -183,7 +183,7 @@ def test_clientvllm_model_name(
     ) -> None:
         llm = ClientvLLM(
             base_url="http://localhost:8000/v1",
-            tokenizer="meta-llama/Meta-Llama-3.1-8B-Instruct",
+            tokenizer="google-bert/bert-base-uncased",
         )
 
         llm.load()
@@ -201,7 +201,7 @@ async def test_agenerate(
     ) -> None:
         llm = ClientvLLM(
             base_url="http://localhost:8000/v1",
-            tokenizer="meta-llama/Meta-Llama-3.1-8B-Instruct",
+            tokenizer="google-bert/bert-base-uncased",
         )
 
         llm.load()