Zpwang-AI commited on
Commit
0e2359d
1 Parent(s): 4797c32

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +14 -6
requirements.txt CHANGED
@@ -1,6 +1,14 @@
1
- torch
2
- gradio
3
- yaml
4
- lightning
5
- deepspeed
6
- transformers
 
 
 
 
 
 
 
 
 
1
+ deepspeed==0.8.3
2
+ fitlog==0.9.15
3
+ gradio==3.25.0
4
+ lightning==2.0.1
5
+ numpy==1.23.4
6
+ pandas==1.5.3
7
+ pynvml==11.5.0
8
+ PyYAML==6.0
9
+ scikit_learn==1.2.2
10
+ torch==1.13.0
11
+ torcheval==0.0.6
12
+ torchmetrics==0.11.4
13
+ tqdm==4.64.1
14
+ transformers==4.25.1