mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-28 09:08:54 +00:00
Madhukar Mishra
62 lines
1.7 KiB
Docker
62 lines
1.7 KiB
Docker
ARG BASE_IMAGE
|
|
FROM ${BASE_IMAGE}
|
|
|
|
SHELL ["/bin/bash", "-o", "pipefail", "-c"]
|
|
ENV DEBIAN_FRONTEND=noninteractive \
|
|
PYTHONUNBUFFERED=1 \
|
|
PIP_NO_CACHE_DIR=on \
|
|
SHELL=/bin/bash
|
|
|
|
# Create and use the Python venv
|
|
WORKDIR /
|
|
RUN python3 -m venv --system-site-packages /venv
|
|
|
|
# Install torch and xformers
|
|
ARG INDEX_URL
|
|
ARG TORCH_VERSION
|
|
ARG XFORMERS_VERSION
|
|
RUN source /venv/bin/activate && \
|
|
pip3 install --no-cache-dir torch==${TORCH_VERSION} torchvision torchaudio --index-url ${INDEX_URL} && \
|
|
pip3 install --no-cache-dir xformers==${XFORMERS_VERSION} --index-url ${INDEX_URL} && \
|
|
deactivate
|
|
|
|
# Clone the git repo of hallo and set version
|
|
ARG HALLO_COMMIT
|
|
RUN git clone https://github.com/fudan-generative-vision/hallo.git && \
|
|
cd /hallo && \
|
|
git checkout ${HALLO_VERSION}
|
|
|
|
# Install the dependencies for hallo
|
|
WORKDIR /hallo
|
|
RUN source /venv/bin/activate && \
|
|
pip3 install -r requirements.txt \
|
|
--extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ --extra-index-url https://download.pytorch.org/whl/cu121/ && \
|
|
pip3 install . && \
|
|
pip3 install huggingface_hub && \
|
|
deactivate
|
|
|
|
# Install models
|
|
RUN source /venv/bin/activate && \
|
|
huggingface-cli download fudan-generative-ai/hallo \
|
|
--local-dir pretrained_models \
|
|
--local-dir-use-symlinks False && \
|
|
deactivate
|
|
|
|
# Remove existing SSH host keys
|
|
RUN rm -f /etc/ssh/ssh_host_*
|
|
|
|
# NGINX Proxy
|
|
COPY nginx/nginx.conf /etc/nginx/nginx.conf
|
|
|
|
# Set template version
|
|
ARG RELEASE
|
|
ENV TEMPLATE_VERSION=${RELEASE}
|
|
|
|
# Copy the scripts
|
|
WORKDIR /
|
|
COPY --chmod=755 scripts/* ./
|
|
|
|
# Start the container
|
|
SHELL ["/bin/bash", "--login", "-c"]
|
|
CMD [ "/start.sh" ]
|