From e03fde3345398f5fcc6813e7a13cedb94f84a706 Mon Sep 17 00:00:00 2001 From: anakin87 Date: Fri, 12 Jan 2024 13:19:36 +0100 Subject: [PATCH] rename model_name_or_path to model in test --- test/components/generators/test_hugging_face_local_generator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/components/generators/test_hugging_face_local_generator.py b/test/components/generators/test_hugging_face_local_generator.py index 46ab153d8e..6cc9844747 100644 --- a/test/components/generators/test_hugging_face_local_generator.py +++ b/test/components/generators/test_hugging_face_local_generator.py @@ -184,7 +184,7 @@ def test_to_dict_with_parameters(self): def test_to_dict_with_quantization_config(self): component = HuggingFaceLocalGenerator( - model_name_or_path="gpt2", + model="gpt2", task="text-generation", device="cuda:0", token="test-token",