jenkins1122 commited on
Commit
8b1d463
·
verified ·
1 Parent(s): 9587959

Upload Dockerfile with huggingface_hub

Browse files
Files changed (1) hide show
  1. Dockerfile +38 -0
Dockerfile ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # SN59 babelbit ARENA miner — gist S2S (whisper-large-v3 → Qwen2.5-7B gist → Kokoro)
2
+ # Validated qualifying score 0.7397 (place #1). Models are DOWNLOADED AT STARTUP (not baked)
3
+ # to keep the image small enough for the arena registry size check (~8 GB vs 28 GB baked).
4
+ # Server self-warms (downloads + loads whisper+Qwen+Kokoro) at startup; serves /v1/predict + /predict + health.
5
+ FROM nvidia/cuda:12.6.0-cudnn-runtime-ubuntu22.04
6
+
7
+ ENV DEBIAN_FRONTEND=noninteractive PIP_NO_CACHE_DIR=1 \
8
+ HF_HOME=/models/hf PIP_BREAK_SYSTEM_PACKAGES=1
9
+
10
+ RUN apt-get update && apt-get install -y --no-install-recommends \
11
+ python3 python3-pip espeak-ng ffmpeg ca-certificates && \
12
+ rm -rf /var/lib/apt/lists/*
13
+
14
+ # torch (CUDA 12.x) + serving + model deps (gist pipeline)
15
+ RUN pip3 install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cu124 && \
16
+ pip3 install --no-cache-dir \
17
+ faster-whisper transformers accelerate bitsandbytes sentencepiece \
18
+ kokoro "misaki[en]" espeakng-loader phonemizer-fork \
19
+ soundfile numpy huggingface_hub fastapi "uvicorn[standard]"
20
+
21
+ WORKDIR /app
22
+
23
+ # Models are NOT baked (would make the image ~28 GB and fail the arena size check).
24
+ # whisper-large-v3 + Qwen2.5-7B + Kokoro-82M download at startup into HF_HOME on first warm.
25
+ RUN mkdir -p /models/hf
26
+
27
+ COPY server.py model.py gist_model.py /app/
28
+
29
+ # Winning config (validated). Managed/arena predict path is /predict; health is GET /health.
30
+ # Babelbit arena contract: port 8000, serves /v1/predict + /predict + /healthz + /health.
31
+ ENV BB_MINER_MODEL=gist \
32
+ BB_ASR_MODEL=large-v3 BB_ASR_BEAMS=5 \
33
+ BB_GIST_LLM=Qwen/Qwen2.5-7B-Instruct BB_GIST_SPEED=1.15 BB_GIST_VOICE=af_heart \
34
+ PORT=8000
35
+
36
+ EXPOSE 8000
37
+ # Server self-warms (loads whisper+Qwen+Kokoro) on startup before accepting traffic.
38
+ CMD ["sh","-c","uvicorn server:app --host 0.0.0.0 --port ${PORT} --log-level info"]