Spaces:
Runtime error
Runtime error
| FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04 | |
| ENV DEBIAN_FRONTEND=noninteractive | |
| RUN apt-get update && \ | |
| apt-get upgrade -y && \ | |
| apt-get install -y --no-install-recommends \ | |
| git \ | |
| git-lfs \ | |
| wget \ | |
| curl \ | |
| # ffmpeg \ | |
| ffmpeg \ | |
| x264 \ | |
| # python build dependencies \ | |
| build-essential \ | |
| libssl-dev \ | |
| zlib1g-dev \ | |
| libbz2-dev \ | |
| libreadline-dev \ | |
| libsqlite3-dev \ | |
| libncursesw5-dev \ | |
| xz-utils \ | |
| tk-dev \ | |
| libxml2-dev \ | |
| libxmlsec1-dev \ | |
| libffi-dev \ | |
| liblzma-dev && \ | |
| apt-get clean && \ | |
| rm -rf /var/lib/apt/lists/* | |
| RUN useradd -m -u 1000 user | |
| USER user | |
| ENV HOME=/home/user \ | |
| PATH=/home/user/.local/bin:${PATH} | |
| WORKDIR ${HOME}/app | |
| RUN curl https://pyenv.run | bash | |
| ENV PATH=${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH} | |
| ENV PYTHON_VERSION=3.10.9 | |
| RUN pyenv install ${PYTHON_VERSION} && \ | |
| pyenv global ${PYTHON_VERSION} && \ | |
| pyenv rehash && \ | |
| pip install --no-cache-dir -U pip setuptools wheel | |
| # RUN pip install --no-cache-dir -U torch==1.13.1 torchvision==0.14.1 | |
| RUN pip install --no-cache-dir -U torch==2.0.0+cu117 torchvision==0.15.1+cu117 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu117 | |
| RUN pip install --no-cache-dir -U xformers==0.0.17 | |
| COPY --chown=1000 requirements.txt /tmp/requirements.txt | |
| RUN pip install --no-cache-dir -U -r /tmp/requirements.txt | |
| COPY --chown=1000 requirements.txt /tmp/requirements.txt | |
| # WORKDIR ${HOME}/app/Make-A-Protagonist/experts/GroundedSAM/segment_anything | |
| # RUN pip install dist/segment_anything-1.0-py3-none-any.whl | |
| # WORKDIR ${HOME}/app/Make-A-Protagonist/experts/GroundedSAM/GroundingDINO | |
| # RUN pip install dist/groundingdino-0.1.0-cp310-cp310-linux_x86_64.whl | |
| # WORKDIR ${HOME}/app | |
| COPY --chown=1000 checkpoints/*.whl /tmp/ | |
| RUN pip install --no-cache-dir -U /tmp/groundingdino-0.1.0-cp310-cp310-linux_x86_64.whl | |
| RUN pip install --no-cache-dir -U /tmp/segment_anything-1.0-py3-none-any.whl | |
| # RUN pip install -e Make-A-Protagonist/experts/GroundedSAM/segment_anything | |
| # RUN pip install -e Make-A-Protagonist/experts/GroundedSAM/GroundingDINO | |
| COPY --chown=1000 . ${HOME}/app | |
| # RUN cd Make-A-Protagonist && patch -p1 < ../patch | |
| ENV PYTHONPATH=${HOME}/app \ | |
| PYTHONUNBUFFERED=1 \ | |
| GRADIO_ALLOW_FLAGGING=never \ | |
| GRADIO_NUM_PORTS=1 \ | |
| GRADIO_SERVER_NAME=0.0.0.0 \ | |
| GRADIO_THEME=huggingface \ | |
| SYSTEM=spaces | |
| CMD ["python", "app.py"] | |