Spaces:
Runtime error
Runtime error
| FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04 | |
| # Configure environment | |
| ENV DEBIAN_FRONTEND=noninteractive \ | |
| PIP_PREFER_BINARY=1 \ | |
| CUDA_HOME=/usr/local/cuda-12.1 \ | |
| TORCH_CUDA_ARCH_LIST="8.6" | |
| # Redirect shell | |
| RUN rm /bin/sh && ln -s /bin/bash /bin/sh | |
| # Install prereqs | |
| RUN apt-get update && apt-get install -y --no-install-recommends \ | |
| curl \ | |
| git-lfs \ | |
| ffmpeg \ | |
| libgl1-mesa-dev \ | |
| libglib2.0-0 \ | |
| git \ | |
| python3-dev \ | |
| python3-pip \ | |
| # Lunar Tools prereqs | |
| libasound2-dev \ | |
| libportaudio2 \ | |
| && apt clean && rm -rf /var/lib/apt/lists/* \ | |
| && ln -s /usr/bin/python3 /usr/bin/python | |
| # Set symbolic links | |
| RUN echo "export PATH=/usr/local/cuda/bin:$PATH" >> /etc/bash.bashrc \ | |
| && echo "export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH" >> /etc/bash. bashrc \ | |
| && echo "export CUDA_HOME=/usr/local/cuda-12.1" >> /etc/bash.bashrc | |
| # Install Python packages: Basic, then CUDA-compatible, then custom | |
| RUN pip3 install \ | |
| wheel \ | |
| ninja && \ | |
| pip3 install \ | |
| torch==2.1.0 \ | |
| torchvision==0.16.0 \ | |
| xformers>=0.0.22 \ | |
| triton>=2.1.0 \ | |
| --index-url https://download.pytorch.org/whl/cu121 && \ | |
| pip3 install git+https://github.com/lunarring/latentblending \ | |
| git+https://github.com/chengzeyi/stable-fast.git@main#egg=stable-fast | |
| # Optionally store weights in image | |
| # RUN mkdir -p /root/.cache/torch/hub/checkpoints/ && curl -o /root/.cache/torch/hub/checkpoints//alexnet-owt-7be5be79.pth https://download.pytorch.org/models/alexnet-owt-7be5be79.pth | |
| # RUN git lfs install && git clone https://huggingface.co/stabilityai/sdxl-turbo /sdxl-turbo | |
| # Clone base repo because why not | |
| RUN git clone https://github.com/lunarring/latentblending.git | |