cog.yaml

build:
  gpu: true

  system_packages:
    - "libgl1-mesa-glx"
    - "libglib2.0-0"

  python_version: "3.8"

  python_packages:
    - "torch==1.7.0"
    - "torchvision==0.8.1"
    - "numpy==1.19.2"
    - "albumentations==0.4.3"
    - "pudb==2019.2"
    - "opencv-python==4.6.0.66"
    - "imageio==2.9.0"
    - "imageio-ffmpeg==0.4.2"
    - "pytorch-lightning==1.5"
    - "omegaconf==2.1.1"
    - "test-tube==0.7.5"
    - "streamlit==0.73.1"
    - "einops==0.3.0"
    - "torch-fidelity==0.3.0"
    - "transformers==4.3.1"
    - "tqdm==4.64.0"
    - "websocket-client==1.3.3"
    - "rel==0.4.7"

  run:
   - "git clone https://github.com/CompVis/taming-transformers.git && cd taming-transformers && pip install -e . && cd .."
   - "git clone https://github.com/openai/CLIP.git && cd CLIP && pip install -e . && cd .."
   - "mkdir -p /root/.cache/ldm/text2img-large/; wget --quiet -O /root/.cache/ldm/text2img-large/model.ckpt https://models.nmb.ai/majesty/latent_diffusion_txt2img_f8_large.ckpt"

image: "r8.im/nightmareai/latent-viz"
predict: "predict-viz.py:Predictor"