feng2022 commited on
Commit
859b33c
1 Parent(s): df52311

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +9 -4
requirements.txt CHANGED
@@ -23,7 +23,12 @@ scipy
23
  matplotlib
24
  pprintpp
25
 
26
- torch==1.11.0
27
- torchvision==0.12.0
28
- torchaudio==0.11.0
29
- cudatoolkit==11.3
 
 
 
 
 
 
23
  matplotlib
24
  pprintpp
25
 
26
+ transformers
27
+ datasets
28
+ accelerate
29
+ nvidia-ml-py3
30
+
31
+ -f https://download.pytorch.org/whl/cu116
32
+ torch
33
+ torchvision
34
+ torchaudio