Add isolated HeartMuLa 3B quality test

This commit is contained in:
Mikei386 committed 2026-09-10 20:20:51 +02:00
1 parent 7c95dab324
commit fe2a93eeb1
5 files changed
+253 -1

No files matched your search

+22
View File
@@ -0,0 +1,22 @@
FROM pytorch/pytorch:2.9.1-cuda12.8-cudnn9-runtime
ARG HEARTLIB_COMMIT=3783bdb8441f2c298b1e64c8651173aac200361c
RUN apt-get update \
&& apt-get install -y --no-install-recommends git ffmpeg curl \
&& rm -rf /var/lib/apt/lists/*
RUN git clone https://github.com/HeartMuLa/heartlib.git /opt/heartlib \
&& git -C /opt/heartlib checkout "${HEARTLIB_COMMIT}" \
&& pip install --no-cache-dir -e /opt/heartlib \
&& pip install --no-cache-dir gradio==5.49.1 uvicorn==0.35.0
COPY app.py /opt/app/app.py
WORKDIR /opt/app
ENV PYTHONUNBUFFERED=1 \
HEARTMULA_MODEL_PATH=/models/ckpt \
HEARTMULA_OUTPUT_DIR=/output
EXPOSE 7860
CMD ["python", "/opt/app/app.py"]