Spaces:
Build error
Build error
Update requirements.txt
Browse files- requirements.txt +15 -8
requirements.txt
CHANGED
|
@@ -1,11 +1,18 @@
|
|
| 1 |
-
|
| 2 |
-
|
| 3 |
-
torch==2.2.1
|
| 4 |
transformers==4.38.2
|
| 5 |
datasets==2.18.0
|
| 6 |
-
deepspeed==0.13.1
|
| 7 |
-
vllm==0.3.0; sys_platform != 'win32'
|
| 8 |
accelerate==0.27.2
|
| 9 |
-
|
| 10 |
-
|
| 11 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Core packages
|
| 2 |
+
torch==2.2.1 ; sys_platform != 'win32'
|
|
|
|
| 3 |
transformers==4.38.2
|
| 4 |
datasets==2.18.0
|
|
|
|
|
|
|
| 5 |
accelerate==0.27.2
|
| 6 |
+
|
| 7 |
+
# Optimization packages
|
| 8 |
+
deepspeed==0.13.1 ; sys_platform != 'win32'
|
| 9 |
+
vllm==0.3.0 ; sys_platform != 'win32'
|
| 10 |
+
unsloth==0.2.0 ; sys_platform != 'win32'
|
| 11 |
+
|
| 12 |
+
# API packages
|
| 13 |
+
fastapi==0.109.1
|
| 14 |
+
uvicorn==0.27.0
|
| 15 |
+
|
| 16 |
+
# Build dependencies
|
| 17 |
+
ninja==1.11.1
|
| 18 |
+
huggingface_hub==0.20.3
|