cog.yaml

build:
  gpu: false
  python_version: "3.10"
  run:
    - apt update -y && apt install -y software-properties-common python3-launchpadlib && apt update -y && add-apt-repository -y ppa:git-core/ppa && apt update -y
    - apt install -y gcc g++ aria2 git git-lfs wget libgl1 libglib2.0-0 ffmpeg cmake libgtk2.0-0 libopenmpi-dev
    - wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run && sh cuda_12.1.0_530.30.02_linux.run --silent --toolkit
    - pip install https://download.pytorch.org/whl/cu121/torch-2.2.1%2Bcu121-cp310-cp310-linux_x86_64.whl
    - pip install https://download.pytorch.org/whl/cu121/xformers-0.0.25-cp310-cp310-manylinux2014_x86_64.whl
    - pip install notebook pyngrok pickleshare ipywidgets imageio-ffmpeg pandas numpy==1.23.5 opencv-python
    - pip install transformers diffusers accelerate ninja colossalai mmengine einops kornia
    - pip install https://github.com/camenduru/wheels/releases/download/colab2/apex-0.1-cp310-cp310-linux_x86_64.whl
    - pip install https://github.com/camenduru/wheels/releases/download/colab2/flash_attn-2.5.6-cp310-cp310-linux_x86_64.whl
    - git clone -b dev https://github.com/camenduru/Open-Sora /content/Open-Sora && cd /content/Open-Sora && pip install -v -e .
    - aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/hpcai-tech/Open-Sora/resolve/main/OpenSora-v1-HQ-16x512x512.pth -d /content/Open-Sora/models -o OpenSora-v1-HQ-16x512x512.pth
    - git clone https://huggingface.co/DeepFloyd/t5-v1_1-xxl /content/Open-Sora/pretrained_models/t5_ckpts/t5-v1_1-xxl
predict: "/src/predict.py:Predictor"