Skip to content

Commit 6737cbd

Browse files
committed
TF ver 4.55.0, pytorch 2.7.1, hf hub 0.34.0 and diffusers 0.31.0
Signed-off-by: Hem Agnihotri <[email protected]>
1 parent 9d9e44a commit 6737cbd

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

pyproject.toml

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,8 @@ classifiers = [
1919
]
2020
requires-python = ">=3.8,<3.11"
2121
dependencies = [
22-
"transformers==4.51.3",
23-
"huggingface-hub==0.30.0",
22+
"transformers==4.55.0",
23+
"huggingface-hub==0.34.0",
2424
"hf_transfer==0.1.9",
2525
"peft==0.13.2",
2626
"datasets==2.20.0",
@@ -39,17 +39,18 @@ dependencies = [
3939
"fire",
4040
"py7zr",
4141
"torchmetrics==1.7.0",
42-
"torch==2.4.1; platform_machine=='aarch64'",
42+
"torch==2.7.1; platform_machine=='aarch64'",
4343
# Specifying torch cpu package URL per python version, update the list once pytorch releases whl for python>3.11
4444
"torch@https://download.pytorch.org/whl/cpu/torch-2.4.1%2Bcpu-cp38-cp38-linux_x86_64.whl ; python_version=='3.8' and platform_machine=='x86_64'",
45-
"torch@https://download.pytorch.org/whl/cpu/torch-2.4.1%2Bcpu-cp39-cp39-linux_x86_64.whl ; python_version=='3.9' and platform_machine=='x86_64'",
46-
"torch@https://download.pytorch.org/whl/cpu/torch-2.4.1%2Bcpu-cp310-cp310-linux_x86_64.whl ; python_version=='3.10' and platform_machine=='x86_64'",
45+
"torch@https://download.pytorch.org/whl/cpu/torch-2.7.1%2Bcpu-cp39-cp39-manylinux_2_28_x86_64.whl ; python_version=='3.9' and platform_machine=='x86_64'",
46+
"torch@https://download.pytorch.org/whl/cpu/torch-2.7.1%2Bcpu-cp310-cp310-manylinux_2_28_x86_64.whl ; python_version=='3.10' and platform_machine=='x86_64'",
4747
]
4848

4949
[project.optional-dependencies]
5050
test = ["pytest","pytest-mock"]
5151
docs = ["Sphinx==7.1.2","sphinx-rtd-theme==2.0.0","myst-parser==3.0.1","sphinx-multiversion"]
5252
quality = ["black", "ruff", "hf_doc_builder@git+https://github.com/huggingface/doc-builder.git"]
53+
diffusers = ["diffusers== 0.31.0"]
5354

5455
[build-system]
5556
requires = ["setuptools>=62.0.0"]

0 commit comments

Comments
 (0)