Content-Length: 1496 | pFad | http://github.com/postgresml/postgresml/pull/1323.patch

thub.com From 0aff2742785e53936885dd0986c0cae4cfce46c2 Mon Sep 17 00:00:00 2001 From: Lev Date: Wed, 21 Feb 2024 14:23:15 -0800 Subject: [PATCH] Bump transformers. Fix HF token --- pgml-extension/requirements.linux.txt | 2 +- pgml-extension/src/bindings/transformers/transformers.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pgml-extension/requirements.linux.txt b/pgml-extension/requirements.linux.txt index 067036d25..3c82504b1 100644 --- a/pgml-extension/requirements.linux.txt +++ b/pgml-extension/requirements.linux.txt @@ -110,7 +110,7 @@ torch==2.1.2 torchaudio==2.1.2 torchvision==0.16.2 tqdm==4.66.1 -transformers==4.36.2 +transformers==4.38.0 transformers-stream-generator==0.0.4 triton==2.1.0 typing-inspect==0.9.0 diff --git a/pgml-extension/src/bindings/transformers/transformers.py b/pgml-extension/src/bindings/transformers/transformers.py index 7bcecc8cc..fadde8858 100644 --- a/pgml-extension/src/bindings/transformers/transformers.py +++ b/pgml-extension/src/bindings/transformers/transformers.py @@ -292,7 +292,7 @@ def __init__(self, model_name, **kwargs): if "token" in kwargs: self.tokenizer = AutoTokenizer.from_pretrained( - model_name, use_auth_token=kwargs["token"] + model_name, token=kwargs["token"] ) else: self.tokenizer = AutoTokenizer.from_pretrained(model_name)

pFad - (p)hone/(F)rame/(a)nonymizer/(d)eclutterfier! Saves Data!