SreyanG-NVIDIA commited on
Commit
7aa36aa
·
verified ·
1 Parent(s): 7aad3a3

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -3
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  # Core dependencies from pyproject.toml
2
- torch==2.3.0
3
- torchvision==0.18.0
4
  transformers==4.46.0
5
  tokenizers>=0.15.2
6
  sentencepiece==0.1.99
@@ -45,7 +45,6 @@ git+https://github.com/bfshi/scaling_on_scales#egg=s2wrapper
45
 
46
  # Additional dependencies from shell script
47
  flash_attn==2.7.3
48
- torchaudio==2.3
49
  soundfile
50
  librosa
51
  openai-whisper
 
1
  # Core dependencies from pyproject.toml
2
+ torch==2.7.0
3
+ torchvision==0.15.2
4
  transformers==4.46.0
5
  tokenizers>=0.15.2
6
  sentencepiece==0.1.99
 
45
 
46
  # Additional dependencies from shell script
47
  flash_attn==2.7.3
 
48
  soundfile
49
  librosa
50
  openai-whisper