Minor correction.

ilya-lavrenov · May 13, 2024 · b106542 · b106542
1 parent 14d8e78
commit b106542
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/text_generation/causal_lm/cpp/continuous_batching/python/tests/test_sampling.py b/text_generation/causal_lm/cpp/continuous_batching/python/tests/test_sampling.py
@@ -95,7 +95,7 @@ def run_hugging_face(
     use_optimum: bool,
     tmp_path: Path
 ) -> Tuple[List[GenerationResult], str]:
-    hf_tokenizer = AutoTokenizer.from_pretrained(model_id, token=os.environ['MODEL_TOKEN'])
+    hf_tokenizer = AutoTokenizer.from_pretrained(model_id)
     model = OVModelForCausalLM.from_pretrained(model_id, export=True) if use_optimum else \
             AutoModelForCausalLM.from_pretrained(model_id)
     generation_results: List[GenerationResult] = []
@@ -196,4 +196,4 @@ def test_hf_models_precommit(tmp_path, model_id):
 @pytest.mark.nightly
 @pytest.mark.parametrize("model_id", get_models_list("models/nightly"))
 def test_hf_models_nightly(tmp_path, model_id):
-    run_test_pipeline(tmp_path, model_id)
+    run_test_pipeline(tmp_path, model_id)