diff --git a/pyproject.toml b/pyproject.toml index 0b02d0a..a6ab34a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -22,7 +22,7 @@ experiments = [ "pycocotools==2.0.11", "PyYAML==6.0.3", "safetensors==0.8.0", - "torch==2.13.0", + "torch==2.14.0", "torchvision==0.28.0", "transformers==5.16.1", "ultralytics==8.4.137", diff --git a/requirements-experiments.txt b/requirements-experiments.txt index c1f9c6a..ce8dd8a 100644 --- a/requirements-experiments.txt +++ b/requirements-experiments.txt @@ -3,7 +3,7 @@ psutil==7.2.2 pycocotools==2.0.11 PyYAML==6.0.3 safetensors==0.8.0 -torch==2.13.0 +torch==2.14.0 torchvision==0.28.0 transformers==5.16.1 ultralytics==8.4.137