LukasHug commited on
Commit
8532f78
·
verified ·
1 Parent(s): c66f1b5

Update requirements.txt

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