cog.yaml

# Configuration for Cog ⚙️
# Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md

build:
  gpu: true
  python_version: "3.9"

  python_packages:
    - "Pillow==10.1.0"
    - "PyYAML==6.0.1"
    - "accelerate==0.24.1"
    - "diffusers==0.18.2"
    - "fire==0.5.0"
    - "git+https://github.com/openai/point-e.git"
    - "gradio==4.8.0"
    - "huggingface_hub==0.16.4"
    - "imageio-ffmpeg==0.4.9"
    - "imageio==2.31.1"
    - "mediapipe==0.10.8"
    - "pip==23.3.1"
    - "plyfile==0.8.1"
    - "safetensors==0.4.0"
    - "torch==2.0.1"
    - "torchaudio==2.0.2"
    - "torchvision==0.15.2"
    - "tqdm==4.66.1"
    - "transformers==4.30.2"
    - "triton==2.0.0"
    - "wandb==0.16.0"
    - "xformers==0.0.20"

  run:
    - curl -o /usr/local/bin/pget -L "https://github.com/replicate/pget/releases/download/v0.5.0/pget" 
    - chmod +x /usr/local/bin/pget
predict: "predict.py:Predictor"

image: r8.im/jd7h/luciddreamer