lorocksUMD commited on
Commit
c23f01b
·
verified ·
1 Parent(s): b758b54

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +28 -1
requirements.txt CHANGED
@@ -1 +1,28 @@
1
- huggingface_hub==0.25.2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ huggingface_hub==0.25.2
2
+ --extra-index-url https://download.pytorch.org/whl/cu118
3
+ pip
4
+ einops
5
+ fastapi
6
+ gradio==3.35.2
7
+ markdown2[all]
8
+ numpy
9
+ requests
10
+ sentencepiece
11
+ tokenizers>=0.12.1
12
+ torch==2.0.1
13
+ torchvision==0.15.2
14
+ uvicorn
15
+ wandb
16
+ shortuuid
17
+ httpx==0.24.0
18
+ deepspeed==0.9.5
19
+ peft==0.4.0
20
+ transformers==4.31.0
21
+ accelerate==0.21.0
22
+ bitsandbytes==0.41.0
23
+ scikit-learn==1.2.2
24
+ sentencepiece==0.1.99
25
+ einops==0.6.1
26
+ einops-exts==0.0.4
27
+ timm==0.6.13
28
+ gradio_client==0.2.9