cog.yaml

# image: "r8.im/laion-ai/ldm-autoedit"
image: "r8.im/laion-ai/erlich"
# image: "r8.im/laion-ai/erlich-autoedit"
# image: "r8.im/laion-ai/ongo"
# image: "r8.im/laion-ai/ongo-autoedit"
build:
  gpu: true

  system_packages:
    - "libopenmpi-dev"
    - "lzma-dev"
    - "liblzma-dev"
    - "software-properties-common"
    - "build-essential"
    - "libnss3-dev"
    - "zlib1g-dev"
    - "libgdbm-dev"
    - "libncurses5-dev"
    - "libssl-dev"
    - "libffi-dev"
    - "libreadline-dev"
    - "libsqlite3-dev"
    - "libbz2-dev"


  python_version: "3.8"

  python_packages:
    - "axial-positional-embedding==0.2.1"
    - "albumentations==0.4.3"
    - "blobfile==1.2.9"
    - "braceexpand==0.1.7"
    - "cachetools==5.0.0"
    - "DALL-E==0.1"
    - "dalle-pytorch==1.5.2"
    - "einops==0.4.1"
    - "gpustat==0.6.0"
    - "huggingface-hub==0.5.1"
    - "imageio==2.9.0"
    - "imageio-ffmpeg==0.4.2"
    - "omegaconf==2.1.2"
    - "pytorch-lightning==1.6"
    - "PyYAML==6.0"
    - "regex==2022.4.24"
    - "rotary-embedding-torch==0.1.5"
    - "tokenizers==0.12.1"
    - "torch==1.10.1"
    - "torchvision==0.11.2"
    - "torchmetrics==0.8.0"
    - "tqdm==4.64.0"
    - "transformers==4.18.0"
    - "torch-fidelity==0.3.0"
    - "wandb==0.12.17"
  
  run:
    - pip3 install -e "git+https://github.com/CompVis/latent-diffusion.git#egg=latent_diffusion"
    - pip3 install -e "git+https://github.com/CompVis/taming-transformers.git#egg=taming_transformers"
    - python3 -c 'from transformers import BertTokenizerFast; t = BertTokenizerFast.from_pretrained("bert-base-uncased");'
    - wget -P /root/.cache/clip "https://openaipublic.azureedge.net/clip/models/b8cca3fd41ae0c99ba7e8951adf17d267cdb84cd88be6f7c2e0eca1737a03836/ViT-L-14.pt"

# predict: "cog_autoedit.py:Predictor"
predict: "cog_sample.py:Predictor"