ginipick commited on
Commit
9e05388
·
verified ·
1 Parent(s): 415626c

Update requirements.txt

Browse files
Files changed (1) hide show
  1. 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
- numpy==2.3.1
8
- pydantic==2.11.7
9
- torch==2.7.0+cu121 # CUDA 12.1 빌드
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
- mamba-ssm==2.2.2 # zonos 모델에서 사용
19
- typing-extensions==4.14.1
 
 
 
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