|
5 | 5 | # We don't declare our dependency on transformers here because we build with |
6 | 6 | # different packages for different variants |
7 | 7 |
|
8 | | -VERSION = "0.5.4" |
| 8 | +VERSION = "0.5.5" |
9 | 9 |
|
10 | 10 | # Ubuntu packages |
11 | 11 | # 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 |
12 | 12 | # ffmpeg: ffmpeg is required for audio processing. On Ubuntu it can be installed as follows: apt install ffmpeg |
13 | 13 | # libavcodec-extra : libavcodec-extra includes additional codecs for ffmpeg |
14 | 14 |
|
15 | 15 | install_requires = [ |
16 | | - "transformers[sklearn,sentencepiece,audio,vision]==4.48.0", |
17 | | - "huggingface_hub[hf_transfer]==0.27.1", |
| 16 | + "transformers[sklearn,sentencepiece,audio,vision]==4.49.0", |
| 17 | + "huggingface_hub[hf_transfer]==0.29.2", |
18 | 18 | # vision |
19 | 19 | "Pillow", |
20 | 20 | "librosa", |
|
31 | 31 |
|
32 | 32 | extras = {} |
33 | 33 |
|
34 | | -extras["st"] = ["sentence_transformers==3.3.1"] |
35 | | -extras["diffusers"] = ["diffusers==0.32.1", "accelerate==1.2.1"] |
| 34 | +extras["st"] = ["sentence_transformers==3.4.1"] |
| 35 | +extras["diffusers"] = ["diffusers==0.32.2", "accelerate==1.4.0"] |
36 | 36 | # Includes `peft` as PEFT requires `torch` so having `peft` as a core dependency |
37 | 37 | # means that `torch` will be installed even if the `torch` extra is not specified. |
38 | 38 | extras["torch"] = ["torch==2.3.1", "torchvision", "torchaudio", "peft==0.14.0"] |
|
57 | 57 | version=VERSION, |
58 | 58 | author="Hugging Face", |
59 | 59 | description="Hugging Face Inference Toolkit is for serving 🤗 Transformers models in containers.", |
60 | | - url="", |
| 60 | + url="https://github.com/huggingface/huggingface-inference-toolkit", |
61 | 61 | package_dir={"": "src"}, |
62 | 62 | packages=find_packages(where="src"), |
63 | 63 | install_requires=install_requires, |
|
0 commit comments