Files changed (1) hide show
  1. requirements.txt +34 -0
requirements.txt ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ git+https://github.com/paralym/LLaVA-NeXT.git@v0.1
2
+ ninja
3
+ opencv-python
4
+ open_clip_torch
5
+ fastapi
6
+ markdown2[all]
7
+ numpy==1.26.4
8
+ requests
9
+ sentencepiece
10
+ torch==2.1.2
11
+ torchvision==0.16.2
12
+ uvicorn
13
+ wandb==0.16.5
14
+ deepspeed==0.12.2
15
+ peft==0.4.0
16
+ accelerate>=0.29.1
17
+ tokenizers~=0.15.2
18
+ transformers
19
+ bitsandbytes==0.41.0
20
+ scikit-learn==1.2.2
21
+ sentencepiece~=0.1.99
22
+ einops==0.6.1
23
+ einops-exts==0.0.4
24
+ pydantic>=2.0
25
+ timm
26
+ hf_transfer
27
+ decord
28
+ datasets
29
+ tyro
30
+ scipy
31
+ rouge
32
+ urllib3~=2.0
33
+ huggingface-hub
34
+ Pillow