YinuoGuo27 commited on
Commit
aa8c6a6
·
1 Parent(s): efc373e

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -3
requirements.txt CHANGED
@@ -1,8 +1,9 @@
1
  # python=3.10
2
  # conda install pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 pytorch-cuda=11.8 -c pytorch -c nvidia
3
- pytorch==2.2.2
4
- torchvision==0.17.2
5
- torchaudio==2.2.2
 
6
  onnxruntime-gpu==1.18.0
7
  transformers==4.33.0
8
  pyyaml==6.0.1
 
1
  # python=3.10
2
  # conda install pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 pytorch-cuda=11.8 -c pytorch -c nvidia
3
+ torch
4
+ torchvision
5
+ torchaudio
6
+ gradio
7
  onnxruntime-gpu==1.18.0
8
  transformers==4.33.0
9
  pyyaml==6.0.1