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
|