Skip to content

Commit 253555b

Browse files
committed
Bump torch, transformers et al.
1 parent 6d7bdb7 commit 253555b

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

setup.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,8 @@
1616
# Due to an error affecting kenlm and cmake (see https://github.com/kpu/kenlm/pull/464)
1717
# Also see the transformers patch for it https://github.com/huggingface/transformers/pull/37091
1818
"kenlm@git+https://github.com/kpu/kenlm@ba83eafdce6553addd885ed3da461bb0d60f8df7",
19-
"transformers[sklearn,sentencepiece,audio,vision]==4.48.0",
20-
"huggingface_hub[hf_transfer]==0.27.1",
19+
"transformers[sklearn,sentencepiece,audio,vision]==4.51.2",
20+
"huggingface_hub[hf_transfer]==0.30.2",
2121
# vision
2222
"Pillow",
2323
"librosa",
@@ -34,11 +34,11 @@
3434

3535
extras = {}
3636

37-
extras["st"] = ["sentence_transformers==3.3.1"]
38-
extras["diffusers"] = ["diffusers==0.32.1", "accelerate==1.2.1"]
37+
extras["st"] = ["sentence_transformers==4.0.2"]
38+
extras["diffusers"] = ["diffusers==0.33.1", "accelerate==1.6.0"]
3939
# Includes `peft` as PEFT requires `torch` so having `peft` as a core dependency
4040
# means that `torch` will be installed even if the `torch` extra is not specified.
41-
extras["torch"] = ["torch==2.3.1", "torchvision", "torchaudio", "peft==0.14.0"]
41+
extras["torch"] = ["torch==2.5.1", "torchvision", "torchaudio", "peft==0.15.1"]
4242
extras["test"] = [
4343
"pytest==7.2.1",
4444
"pytest-xdist",

0 commit comments

Comments
 (0)