yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2026-01-08 04:56:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

rename model_name_or_path to model in test (#6732)

Browse Source
This commit is contained in:
Stefano Fiorucci 2024-01-12 13:56:14 +01:00 committed by GitHub
parent dbdeb8259e
commit 24c71bd221
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/components/generators/test_hugging_face_local_generator.py
View File

@ -184,7 +184,7 @@ class TestHuggingFaceLocalGenerator:
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",