From cdd5e7f2e06789d3f96c9dec9c6b97562c3884e6 Mon Sep 17 00:00:00 2001 From: ZanSara Date: Fri, 12 Jan 2024 14:53:52 +0100 Subject: [PATCH] fix leftover model_name_or_path param --- e2e/pipelines/test_eval_rag_pipelines.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/e2e/pipelines/test_eval_rag_pipelines.py b/e2e/pipelines/test_eval_rag_pipelines.py index 1a9f57f6be..348b5a38fa 100644 --- a/e2e/pipelines/test_eval_rag_pipelines.py +++ b/e2e/pipelines/test_eval_rag_pipelines.py @@ -28,7 +28,7 @@ def test_bm25_rag_pipeline(tmp_path): rag_pipeline.add_component(instance=PromptBuilder(template=prompt_template), name="prompt_builder") rag_pipeline.add_component( instance=HuggingFaceLocalGenerator( - model_name_or_path="google/flan-t5-small", + model="google/flan-t5-small", task="text2text-generation", generation_kwargs={"max_new_tokens": 100, "temperature": 0.5, "do_sample": True}, ), @@ -102,7 +102,7 @@ def test_embedding_retrieval_rag_pipeline(tmp_path): rag_pipeline.add_component(instance=PromptBuilder(template=prompt_template), name="prompt_builder") rag_pipeline.add_component( instance=HuggingFaceLocalGenerator( - model_name_or_path="google/flan-t5-small", + model="google/flan-t5-small", task="text2text-generation", generation_kwargs={"max_new_tokens": 100, "temperature": 0.5, "do_sample": True}, ),