Content-Length: 1118 | pFad | http://github.com/postgresml/postgresml/pull/1323.diff
thub.com 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)Fetched URL: http://github.com/postgresml/postgresml/pull/1323.diff
Alternative Proxies: