Fred808 commited on
Commit
386ac2a
·
verified ·
1 Parent(s): c6f1420

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +15 -7
requirements.txt CHANGED
@@ -1,7 +1,15 @@
1
- torch
2
- transformers
3
- accelerate
4
- fastapi
5
- uvicorn
6
- pillow
7
- protobuf
 
 
 
 
 
 
 
 
 
1
+ torch>=2.1.0
2
+ transformers>=4.40.0
3
+ accelerate>=0.28.0
4
+ fastapi>=0.110.0
5
+ uvicorn[standard]>=0.29.0
6
+ pillow>=10.0.0
7
+ protobuf>=4.25.0
8
+
9
+ # Optional: for better performance and compatibility
10
+ scipy>=1.11.0
11
+ tokenizers>=0.15.0
12
+ requests>=2.31.0
13
+
14
+ # If using CUDA, add the appropriate torch version for your CUDA version
15
+ # See: https://pytorch.org/get-started/locally/