cog.yaml

# Configuration for Cog ⚙️
# Reference: https://cog.run/yaml

build:
  gpu: true
  cuda: "12.4"
  python_version: "3.10.14"
  python_packages:
    # flux
    - "torch==2.4.1"
    - "torchvision==0.19.1"
    - "safetensors==0.4.4"
    - "diffusers==0.30.0"
    - "transformers==4.44.0"
    - "lycoris-lora==1.8.3"
    - "flatten_json==0.1.14"
    - "pyyaml==6.0.1"
    - "oyaml==1.0"
    - "tensorboard==2.17.0"
    - "kornia==0.7.3"
    - "einops==0.8.0"
    - "accelerate==0.33.0"
    - "toml==0.10.2"
    - "albumentations==1.4.3"
    - "opencv-python==4.10.0.84"
    - "pillow==10.4.0"
    - "pydantic==1.10.17"
    - "omegaconf==2.3.0"
    - "k-diffusion"
    - "open_clip_torch==2.26.1"
    - "timm==1.0.8"
    - "prodigyopt==1.0"
    - "controlnet_aux==0.0.7"
    - "bitsandbytes==0.43.3"
    - "scikit-image==0.24.0"
    - "huggingface-hub==0.24.5"
    - "hf_transfer==0.1.8"
    - "lpips==0.1.4"
    - "optimum-quanto==0.2.4"
    - "sentencepiece==0.2.0"
    - "peft==0.13.0"
    - "lion-pytorch==0.2.2"

    # llava
    - "deepspeed==0.9.5"
    - "einops-exts==0.0.4"
    - "gradio==3.35.2"
    - "gradio_client==0.2.9"
    - "httpx==0.24.0"
    - "markdown2==2.4.10"
    - "numpy==1.26.0"
    - "shortuuid==1.0.11"
    - "tokenizers==0.19"
    - "wandb==0.17.8"
    - "wavedrom==2.0.3.post3"
    - "Pygments==2.16.1"
  run:
    - curl -o /usr/local/bin/pget -L "https://github.com/replicate/pget/releases/download/v0.8.2/pget_linux_x86_64" && chmod +x /usr/local/bin/pget
    - pip install "git+https://github.com/Gothos/diffusers.git@flux-inpaint"
    # - "pip install --force-reinstall --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu124"
    # - "pip install --force-reinstall --pre torch==2.5.0.dev20240905+cu124 torchvision==0.20.0.dev20240905+cu124 --index-url https://download.pytorch.org/whl/nightly/cu124"

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