Koaris commited on
Commit
67c64c1
·
verified ·
1 Parent(s): 41a4b25

Create Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +30 -0
Dockerfile ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM nvidia/cuda:12.1.1-runtime-ubuntu20.04
2
+
3
+ # Install basic system deps
4
+ RUN apt-get update && apt-get install -y \
5
+ git python3 python3-pip ffmpeg libsndfile1 \
6
+ && rm -rf /var/lib/apt/lists/*
7
+
8
+ # Install Python deps
9
+ RUN pip install --upgrade pip
10
+
11
+ # Install nightly vLLM with audio support
12
+ RUN pip install "vllm[audio]" --extra-index-url https://wheels.vllm.ai/nightly
13
+
14
+ # Optional: mistral_common should come with vllm[audio]
15
+ RUN pip install mistral_common[audio] openai gradio huggingface_hub
16
+
17
+ # Expose the default Gradio port
18
+ EXPOSE 7860
19
+
20
+ # Set environment variables
21
+ ENV HF_HUB_ENABLE_HF_TRANSFER=1
22
+
23
+ # Download and launch model + Gradio client
24
+ COPY app.py .
25
+
26
+ CMD vllm.serve mistralai/Voxtral-Mini-3B-2507 \
27
+ --tokenizer_mode mistral \
28
+ --config_format mistral \
29
+ --load_format mistral & \
30
+ python3 app.py