diff --git a/benchmark/requirements.txt b/benchmark/requirements.txt index 50e9dfaddfa4..e17358667e1c 100644 --- a/benchmark/requirements.txt +++ b/benchmark/requirements.txt @@ -1,5 +1,5 @@ gpustat==1.1.1 psutil==6.0.0 psycopg2==2.9.9 -torch>=2.4.0 +torch>=2.13.0 hf_transfer \ No newline at end of file diff --git a/examples/flax/vision/requirements.txt b/examples/flax/vision/requirements.txt index 116eab862910..403c6367c4ea 100644 --- a/examples/flax/vision/requirements.txt +++ b/examples/flax/vision/requirements.txt @@ -3,6 +3,6 @@ jaxlib>=0.1.59 flax>=0.3.5 optax>=0.0.8 -f https://download.pytorch.org/whl/torch_stable.html -torch==2.7.0 +torch==2.12.1 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.12.0+cpu diff --git a/examples/legacy/benchmarking/requirements.txt b/examples/legacy/benchmarking/requirements.txt index 68c56b321909..1efbc4433571 100644 --- a/examples/legacy/benchmarking/requirements.txt +++ b/examples/legacy/benchmarking/requirements.txt @@ -1 +1 @@ -torch >= 1.3 \ No newline at end of file +torch >= 2.13.0 \ No newline at end of file diff --git a/examples/pytorch/audio-classification/requirements.txt b/examples/pytorch/audio-classification/requirements.txt index acf058d4cf46..1aed1632ea65 100644 --- a/examples/pytorch/audio-classification/requirements.txt +++ b/examples/pytorch/audio-classification/requirements.txt @@ -2,4 +2,4 @@ datasets>=1.14.0 evaluate librosa torchaudio -torch>=1.6 \ No newline at end of file +torch>=2.13.0 \ No newline at end of file diff --git a/examples/tensorflow/language-modeling-tpu/requirements.txt b/examples/tensorflow/language-modeling-tpu/requirements.txt index 0c530ff9b578..ab446d923ffb 100644 --- a/examples/tensorflow/language-modeling-tpu/requirements.txt +++ b/examples/tensorflow/language-modeling-tpu/requirements.txt @@ -1,3 +1,3 @@ -transformers==4.50.0 +transformers==5.5.0 datasets==2.9.0 tokenizers==0.13.2