kaifz commited on
Commit
497691a
·
1 Parent(s): 8e6b993
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -23,6 +23,6 @@ timm
23
  omegaconf
24
  hydra-core
25
  wandb
26
- torch==2.4.1 --index-url https://download.pytorch.org/whl/cu124
27
- torchvision==0.19.1 --index-url https://download.pytorch.org/whl/cu124
28
- torchaudio==2.4.1 --index-url https://download.pytorch.org/whl/cu124
 
23
  omegaconf
24
  hydra-core
25
  wandb
26
+ torch==2.6.0 --index-url https://download.pytorch.org/whl/cu124
27
+ torchvision==0.21.0 --index-url https://download.pytorch.org/whl/cu124
28
+ torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu124