cog.yaml

build:
  # set to true if your model requires a GPU
  gpu: true

  system_packages:
    - ffmpeg

  python_version: "3.10.6"

  python_packages:
    - torch
    - torchvision
    - torchaudio
    - torchsde
    - pandas
    - einops
    - transformers>=4.25.1
    - safetensors>=0.3.0
    - opencv-python
    - aiohttp
    - accelerate
    - pyyaml
    - Pillow
    - scipy
    - imageio-ffmpeg
    - scikit-image
    - tqdm
    - numexpr
    - psutil
    - matplotlib
    - websocket-client==1.6.3
    - "tensorflow_addons==0.23.0" # needed for Bumble nsfw detection model
    - open_clip_torch   # needed for CLIP_interrogator node

  pre_install:
    - pip install tensorflow==2.11.0

# predict.py defines how predictions are run on your model
predict: "predict.py:Predictor"

image: "r8.im/abraham-ai/eden-comfyui"