FROM runpod/pytorch:2.2.1-py3.10-cuda12.1.1-devel-ubuntu22.04
WORKDIR /content
ENV PATH="/home/zebraslive/.local/bin:${PATH}"

RUN adduser --disabled-password --gecos '' zebraslive && \
    adduser zebraslive sudo && \
    echo '%sudo ALL=(ALL) NOPASSWD:ALL' >> /etc/sudoers && \
    chown -R zebraslive:zebraslive /content && \
    chmod -R 777 /content && \
    chown -R zebraslive:zebraslive /home && \
    chmod -R 777 /home && \
    apt update -y && add-apt-repository -y ppa:git-core/ppa && apt update -y && apt install -y aria2 git git-lfs unzip ffmpeg

USER zebraslive


COPY ./cogvideox /content/cogvideox

RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M "https://cdn-lfs-us-1.hf.co/repos/c9/2c/c92c727c91777ba0dc27765cde8230ff8b79131db869adc443ff9f2b06973eaf/15f88876af8aacc9fe8f88d3c80dbe00e4a7b24e64f28e9b72c2e9551e9720e0?response-content-disposition=attachment%3B+filename*%3DUTF-8%27%271000_latestHD_ain.safetensors%3B+filename%3D%221000_latestHD_ain.safetensors%22%3B&Expires=1731555078&Policy=eyJTdGF0ZW1lbnQiOlt7IkNvbmRpdGlvbiI6eyJEYXRlTGVzc1RoYW4iOnsiQVdTOkVwb2NoVGltZSI6MTczMTU1NTA3OH19LCJSZXNvdXJjZSI6Imh0dHBzOi8vY2RuLWxmcy11cy0xLmhmLmNvL3JlcG9zL2M5LzJjL2M5MmM3MjdjOTE3NzdiYTBkYzI3NzY1Y2RlODIzMGZmOGI3OTEzMWRiODY5YWRjNDQzZmY5ZjJiMDY5NzNlYWYvMTVmODg4NzZhZjhhYWNjOWZlOGY4OGQzYzgwZGJlMDBlNGE3YjI0ZTY0ZjI4ZTliNzJjMmU5NTUxZTk3MjBlMD9yZXNwb25zZS1jb250ZW50LWRpc3Bvc2l0aW9uPSoifV19&Signature=o6WjhiW6KJHrJSMRupTCBzvT1LKJ1E374doRJgI77br9Q9osPkytCfFnebhkcYnUVm5UzPj-tGrSvW-pa3BxHKZaZQFBL4ODaR%7ExYhsVxkMnQDkJCMfhx0fL6LC7zrZkta1ejtUzitmi%7EcGuz2rtgasx0UrB%7EgeA8z7Fo-QD4dcIF06zQAty1OGpTHirN49ImDRdyCYCGce1orMcOzlUiFHNxZYk5Usm63ZUU3wNIfMuwomE9MIxDTah4hxpqGqF6KlqqVok4M0PMf9nJ0PmthQmOlcxZSNxxIwLsASEYtr36KBE0ZBnyWDwNiQAx3DSHqUUuMpZBpdRLQPTjP9uHA__&Key-Pair-Id=K24J24Z295AEI9" -d /content/ -o shirtlift.safetensors

COPY ./worker_runpod.py /content/worker_runpod.py
WORKDIR /content
CMD python worker_runpod.py