Spaces:
Running
on
Zero
Running
on
Zero
Update requirements.txt
Browse files- requirements.txt +9 -13
requirements.txt
CHANGED
|
@@ -1,20 +1,16 @@
|
|
| 1 |
-
# requirements.txt
|
| 2 |
-
aiofiles==23.2.1
|
| 3 |
anyio==4.8.0
|
| 4 |
fastapi==0.116.1
|
| 5 |
gradio==5.16.0
|
| 6 |
httpx==0.28.1
|
| 7 |
-
|
| 8 |
-
|
| 9 |
-
|
| 10 |
-
torchvision==0.22.0+cu121
|
| 11 |
-
torchaudio==2.7.0+cu121
|
| 12 |
-
triton==3.3.0 # torch 2.7.x 요구
|
| 13 |
-
flash-attn==2.8.2 # 최신, wrap_triton 사용
|
| 14 |
-
transformers==4.49.0
|
| 15 |
-
huggingface-hub==0.33.2
|
| 16 |
safetensors==0.5.3
|
| 17 |
soundfile==0.13.1
|
| 18 |
-
|
| 19 |
-
|
|
|
|
|
|
|
|
|
|
| 20 |
uvicorn==0.35.0
|
|
|
|
| 1 |
+
# requirements.txt
|
|
|
|
| 2 |
anyio==4.8.0
|
| 3 |
fastapi==0.116.1
|
| 4 |
gradio==5.16.0
|
| 5 |
httpx==0.28.1
|
| 6 |
+
huggingface-hub==0.28.1
|
| 7 |
+
mamba-ssm==2.2.2
|
| 8 |
+
numpy==2.2.6 # Python 3.10 호환
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
safetensors==0.5.3
|
| 10 |
soundfile==0.13.1
|
| 11 |
+
torch==2.6.0+cu124 # Dockerfile에서 미리 설치
|
| 12 |
+
triton==3.2.0 # torch 2.6.x 요구
|
| 13 |
+
flash-attn==2.8.1 # wrap_triton 사용, torch 2.6+ 호환
|
| 14 |
+
transformers==4.49.0
|
| 15 |
+
typing-extensions==4.12.2
|
| 16 |
uvicorn==0.35.0
|