mit-moeflow-worker / Dockerfile
jokester's picture
update commit id
4fbdb1f
FROM mambaorg/micromamba:2-debian12-slim
COPY . /home/mambauser/bootstrap
RUN --mount=type=cache,uid=57439,gid=57439,target=/opt/conda/pkgs micromamba env create --yes --file /home/mambauser/bootstrap/conda.yaml \
&& micromamba install -n base --yes curl
USER root
# for some stupid reason hugging face run container images as uid=1000
RUN useradd -m -u 1000 runtime-user
WORKDIR /app
RUN chown -Rc runtime-user /app
USER runtime-user
RUN mkdir -pv /app/storage \
&& micromamba run -n mit-py311 python3 -mvenv --system-site-packages /app/venv
RUN venv/bin/pip install -r /home/mambauser/bootstrap/requirements-moeflow.txt
# HF does not seem to have layer cache anyway
# (must specify -n base to use installed curl)
RUN micromamba run -n base curl -L https://github.com/moeflow-com/manga-image-translator/archive/d4c8402f503719b14432cc4831305a0e810cdc2d.tar.gz | tar xvz --strip-components=1
# NOTE for unknown reason we need to download here. or there will be runtime file permission error
RUN venv/bin/python docker_prepare.py --models ocr.48px,ocr.48px_ctc,ocr.32px,ocr.mocr,detector.default,detector.ctd,detector.craft,detector.none
CMD venv/bin/python gradio-multi.py