Eeppa commited on
Commit
ca7274c
·
verified ·
1 Parent(s): d79be6f

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -1
requirements.txt CHANGED
@@ -1,5 +1,7 @@
1
  gradio>=4.0
2
- torch --index-url https://download.pytorch.org/whl/cpu
 
 
3
  transformers
4
  datasets
5
  accelerate
 
1
  gradio>=4.0
2
+ torch==2.3.1+cpu --index-url https://download.pytorch.org/whl/cpu
3
+ torchvision==0.18.1+cpu --index-url https://download.pytorch.org/whl/cpu
4
+ torchaudio==2.3.1+cpu --index-url https://download.pytorch.org/whl/cpu
5
  transformers
6
  datasets
7
  accelerate