refactor: reduce image size

This commit is contained in:
Koro33
2024-04-01 18:32:36 +02:00
parent 70d794a0ad
commit bffc5325ae
2 changed files with 24 additions and 8 deletions

6
.gitignore vendored
View File

@@ -1,6 +1,10 @@
# from .gitignore
venv/ venv/
ui/__pycache__/ ui/__pycache__/
outputs/ outputs/
modules/__pycache__/ modules/__pycache__/
models/ models/
modules/yt_tmp.wav modules/yt_tmp.wav
.git
.github

View File

@@ -1,22 +1,34 @@
FROM nvidia/cuda:12.3.2-cudnn9-runtime-ubuntu22.04 AS runtime FROM debian:bookworm-slim AS builder
VOLUME [ "/Whisper-WebUI/models" ]
VOLUME [ "/Whisper-WebUI/outputs" ]
RUN apt-get update && \ RUN apt-get update && \
apt-get install -y curl ffmpeg git python3 python3-pip python3-venv && \ apt-get install -y curl git python3 python3-pip python3-venv && \
rm -rf /var/lib/apt/lists/* /var/cache/apt/archives/* && \ rm -rf /var/lib/apt/lists/* /var/cache/apt/archives/* && \
mkdir -p /Whisper-WebUI mkdir -p /Whisper-WebUI
WORKDIR /Whisper-WebUI WORKDIR /Whisper-WebUI
COPY . . COPY requirements.txt .
RUN python3 -m venv venv && \ RUN python3 -m venv venv && \
. venv/bin/activate && \ . venv/bin/activate && \
pip install --no-cache-dir -r requirements.txt pip install --no-cache-dir -r requirements.txt
FROM debian:bookworm-slim AS runtime
RUN apt-get update && \
apt-get install -y curl ffmpeg python3 && \
rm -rf /var/lib/apt/lists/* /var/cache/apt/archives/*
WORKDIR /Whisper-WebUI
COPY . .
COPY --from=builder /Whisper-WebUI/venv /Whisper-WebUI/venv
VOLUME [ "/Whisper-WebUI/models" ]
VOLUME [ "/Whisper-WebUI/outputs" ]
ENV PATH="/Whisper-WebUI/venv/bin:$PATH" ENV PATH="/Whisper-WebUI/venv/bin:$PATH"
ENV LD_LIBRARY_PATH=/Whisper-WebUI/venv/lib64/python3.10/site-packages/nvidia/cublas/lib:/Whisper-WebUI/venv/lib64/python3.10/site-packages/nvidia/cudnn/lib ENV LD_LIBRARY_PATH=/Whisper-WebUI/venv/lib64/python3.11/site-packages/nvidia/cublas/lib:/Whisper-WebUI/venv/lib64/python3.11/site-packages/nvidia/cudnn/lib
ENTRYPOINT [ "python", "app.py" ] ENTRYPOINT [ "python", "app.py" ]