upgrade transformers (#5619)

This commit is contained in:
Stefano Fiorucci 2023-08-25 16:38:34 +02:00 committed by GitHub
parent 7c7a486014
commit 8342b6a457
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 2 deletions

View File

@ -48,7 +48,7 @@ classifiers = [
dependencies = [
"requests",
"pydantic<2",
"transformers==4.31.0",
"transformers==4.32.0",
"safetensors<0.3.2", # related issue: https://github.com/huggingface/safetensors/issues/314
"pandas",
"rank_bm25",
@ -90,7 +90,7 @@ dependencies = [
[project.optional-dependencies]
inference = [
"transformers[torch,sentencepiece]==4.31.0",
"transformers[torch,sentencepiece]==4.32.0",
"sentence-transformers>=2.2.0", # See haystack/nodes/retriever/_embedding_encoder.py, _SentenceTransformersEmbeddingEncoder
"huggingface-hub>=0.5.0",
]

View File

@ -0,0 +1,5 @@
---
enhancements:
- |
Upgrade Transformers to the latest version 4.32.0.
This version adds support for the GPTQ quantization and integrates MPT models.