Skip to content

Bump transformers and sentence-transformers #103

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 10 commits into from
Apr 23, 2025
Merged

Bump transformers and sentence-transformers #103

Changes from 3 commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
52fe9eb
Bump `transformers`, `diffusers`, `sentence-transformers` et al.
alvarobartt Mar 10, 2025
ec8d0c7
try just popping from inputs
ErikKaum Mar 10, 2025
9588098
Revert "try just popping from inputs"
ErikKaum Mar 10, 2025
8201938
maybe fix torch dtype issue
ErikKaum Mar 10, 2025
da5a587
maybe we hardcode bfloat16
ErikKaum Mar 10, 2025
f877fbd
try with lower diffusers patch
ErikKaum Mar 10, 2025
dfc1ed3
revers all versions execpt transformers
ErikKaum Mar 10, 2025
893de39
Merge branch 'main' into bump-versions-and-release
alvarobartt Apr 14, 2025
6d7bdb7
Drop Python 3.8 support
alvarobartt Apr 14, 2025
253555b
Bump `torch`, `transformers` et al.
alvarobartt Apr 14, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 6 additions & 6 deletions setup.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,16 +5,16 @@
# We don't declare our dependency on transformers here because we build with
# different packages for different variants

VERSION = "0.5.4"
VERSION = "0.5.5"

# Ubuntu packages
# libsndfile1-dev: torchaudio requires the development version of the libsndfile package which can be installed via a system package manager. On Ubuntu it can be installed as follows: apt install libsndfile1-dev
# ffmpeg: ffmpeg is required for audio processing. On Ubuntu it can be installed as follows: apt install ffmpeg
# libavcodec-extra : libavcodec-extra includes additional codecs for ffmpeg

install_requires = [
"transformers[sklearn,sentencepiece,audio,vision]==4.48.0",
"huggingface_hub[hf_transfer]==0.27.1",
"transformers[sklearn,sentencepiece,audio,vision]==4.49.0",
"huggingface_hub[hf_transfer]==0.29.2",
# vision
"Pillow",
"librosa",
Expand All @@ -31,8 +31,8 @@

extras = {}

extras["st"] = ["sentence_transformers==3.3.1"]
extras["diffusers"] = ["diffusers==0.32.1", "accelerate==1.2.1"]
extras["st"] = ["sentence_transformers==3.4.1"]
extras["diffusers"] = ["diffusers==0.32.2", "accelerate==1.4.0"]
# Includes `peft` as PEFT requires `torch` so having `peft` as a core dependency
# means that `torch` will be installed even if the `torch` extra is not specified.
extras["torch"] = ["torch==2.3.1", "torchvision", "torchaudio", "peft==0.14.0"]
Expand All @@ -57,7 +57,7 @@
version=VERSION,
author="Hugging Face",
description="Hugging Face Inference Toolkit is for serving 🤗 Transformers models in containers.",
url="",
url="https://github.com/huggingface/huggingface-inference-toolkit",
package_dir={"": "src"},
packages=find_packages(where="src"),
install_requires=install_requires,
Expand Down
Loading