hoonsubin commited on
Commit
2fc6bee
·
1 Parent(s): 597e812

add dockerfile

Browse files
Files changed (2) hide show
  1. Dockerfile +31 -0
  2. requirements.txt +31 -33
Dockerfile ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-devel
2
+
3
+ WORKDIR /app
4
+
5
+ # Install system dependencies
6
+ RUN apt-get update && apt-get install -y \
7
+ libportaudio2 \
8
+ libportaudiocpp0 \
9
+ portaudio19-dev \
10
+ libasound-dev \
11
+ libsndfile1-dev \
12
+ kmod \
13
+ build-essential \
14
+ && rm -rf /var/lib/apt/lists/*
15
+
16
+ # Update pip first
17
+ RUN pip install --upgrade pip setuptools wheel
18
+
19
+ # Install base requirements first
20
+ RUN pip install numpy==1.26.4
21
+
22
+ # Install vLLM with CUDA 12.1 support
23
+ RUN pip install vllm==0.6.4.post1 \
24
+ --extra-index-url https://pypi.nvidia.com \
25
+ --extra-index-url https://download.pytorch.org/whl/cu121
26
+
27
+ COPY . .
28
+ RUN pip install -r requirements.txt
29
+
30
+ EXPOSE 7860
31
+ CMD ["python", "app.py"]
requirements.txt CHANGED
@@ -1,5 +1,36 @@
1
  # This file was autogenerated by uv via the following command:
2
  # uv pip compile --output-file requirements.txt pyproject.toml
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  aiofiles==23.2.1
4
  # via
5
  # auralis
@@ -322,30 +353,6 @@ numba==0.61.0
322
  # via
323
  # librosa
324
  # outlines
325
- numpy==1.26.4
326
- # via
327
- # auralis
328
- # blis
329
- # datasets
330
- # gguf
331
- # gradio
332
- # langid
333
- # librosa
334
- # mistral-common
335
- # numba
336
- # opencv-python-headless
337
- # outlines
338
- # pandas
339
- # pyloudnorm
340
- # scikit-learn
341
- # scipy
342
- # soundfile
343
- # soxr
344
- # spacy
345
- # thinc
346
- # torchvision
347
- # transformers
348
- # vllm
349
  nvidia-ml-py==12.570.86
350
  # via
351
  # auralis
@@ -634,13 +641,6 @@ tomli==2.2.1
634
  # via pytest
635
  tomlkit==0.13.2
636
  # via gradio
637
- torch==2.5.1
638
- # via
639
- # auralis-tts (pyproject.toml)
640
- # compressed-tensors
641
- # torchaudio
642
- # torchvision
643
- # vllm
644
  torchaudio==2.5.1
645
  # via
646
  # auralis-tts (pyproject.toml)
@@ -714,8 +714,6 @@ uvicorn==0.34.0
714
  # vllm
715
  uvloop==0.21.0
716
  # via uvicorn
717
- vllm==0.6.4.post1
718
- # via auralis
719
  wasabi==0.10.1
720
  # via
721
  # spacy
 
1
  # This file was autogenerated by uv via the following command:
2
  # uv pip compile --output-file requirements.txt pyproject.toml
3
+ numpy==1.26.4
4
+ # via
5
+ # auralis
6
+ # blis
7
+ # datasets
8
+ # gguf
9
+ # gradio
10
+ # langid
11
+ # librosa
12
+ # mistral-common
13
+ # numba
14
+ # opencv-python-headless
15
+ # outlines
16
+ # pandas
17
+ # pyloudnorm
18
+ # scikit-learn
19
+ # scipy
20
+ # soundfile
21
+ # soxr
22
+ # spacy
23
+ # thinc
24
+ # torchvision
25
+ # transformers
26
+ # vllm
27
+ torch==2.5.1
28
+ # via
29
+ # auralis-tts (pyproject.toml)
30
+ # compressed-tensors
31
+ # torchaudio
32
+ # torchvision
33
+ # vllm
34
  aiofiles==23.2.1
35
  # via
36
  # auralis
 
353
  # via
354
  # librosa
355
  # outlines
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
356
  nvidia-ml-py==12.570.86
357
  # via
358
  # auralis
 
641
  # via pytest
642
  tomlkit==0.13.2
643
  # via gradio
 
 
 
 
 
 
 
644
  torchaudio==2.5.1
645
  # via
646
  # auralis-tts (pyproject.toml)
 
714
  # vllm
715
  uvloop==0.21.0
716
  # via uvicorn
 
 
717
  wasabi==0.10.1
718
  # via
719
  # spacy