Sharaf Zaman commited on
Commit
78d59c2
1 Parent(s): 7946f35

Explicitly set the requirements (llava-torch seems to have some of them outdated..)

Browse files
Files changed (1) hide show
  1. requirements.txt +22 -1
requirements.txt CHANGED
@@ -1 +1,22 @@
1
- llava-torch==1.2.2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ torch==2.1.2
2
+ torchvision==0.16.2
3
+ transformers==4.37.2
4
+ tokenizers==0.15.1
5
+ sentencepiece==0.1.99
6
+ shortuuid
7
+ accelerate==0.21.0
8
+ peft==0.5.0
9
+ bitsandbytes
10
+ pydantic
11
+ markdown2[all]
12
+ numpy
13
+ scikit-learn==1.2.2
14
+ gradio==4.16.0
15
+ gradio_client==0.8.1
16
+ requests
17
+ httpx==0.24.0
18
+ uvicorn
19
+ fastapi
20
+ einops==0.6.1
21
+ einops-exts==0.0.4
22
+ timm==0.6.13