Files
edenartlab-sd-lora-trainer/cog.yaml
T
2024-05-14 15:11:07 -07:00

20 lines
616 B
YAML
Executable File

# Configuration for Cog ⚙️
# Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md
build:
gpu: true
cuda: "11.8"
python_version: "3.9"
system_packages:
- "ffmpeg"
- "libgl1-mesa-glx"
- "libegl1-mesa-dev"
- "libsm6"
- "libxext6"
python_requirements: requirements.txt
run:
- wget http://thegiflibrary.tumblr.com/post/11565547760 -O face_landmarker_v2_with_blendshapes.task -q https://storage.googleapis.com/mediapipe-models/face_landmarker/face_landmarker/float16/1/face_landmarker.task
predict: "predict.py:Predictor"
image: "r8.im/edenartlab/sdxl-lora-trainer"