Add isolated HeartMuLa 3B quality test
This commit is contained in:
1 parent
7c95dab324
commit
fe2a93eeb1
5 files changed
+253
-1
No files matched your search
@@ -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"]
|
||||
Reference in new issue
Block a user