From abbbf0de3ebdf3fb1abd1e427921f2e5cf23cf6e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mateo=20Peci=C3=B1a?= Date: Tue, 5 Dec 2023 13:57:22 +0100 Subject: [PATCH] Change use_auth_token to token in TransformersDocumentClassifier --- haystack/nodes/document_classifier/transformers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/haystack/nodes/document_classifier/transformers.py b/haystack/nodes/document_classifier/transformers.py index f5c2a84f6b..c58be4abb4 100644 --- a/haystack/nodes/document_classifier/transformers.py +++ b/haystack/nodes/document_classifier/transformers.py @@ -149,7 +149,7 @@ def __init__( model=model_name_or_path, tokenizer=tokenizer, revision=model_version, - use_auth_token=use_auth_token, + token=use_auth_token, device=resolved_devices[0], ) elif task == "text-classification": @@ -160,7 +160,7 @@ def __init__( device=resolved_devices[0], revision=model_version, top_k=top_k, - use_auth_token=use_auth_token, + token=use_auth_token, ) self.top_k = top_k self.labels = labels