|
# Configuration for Cog ⚙️ |
|
# Reference: https: |
|
|
|
build: |
|
gpu: true |
|
cuda: "11.4" |
|
python_version: "3.10" |
|
system_packages: |
|
- "ffmpeg" |
|
- "libsm6" |
|
- "libxext6" |
|
python_packages: |
|
- "diffusers==0.11.1" |
|
- "torch==1.13.0" |
|
- "ftfy==6.1.1" |
|
- "scipy==1.9.3" |
|
- "transformers==4.25.1" |
|
- "accelerate==0.15.0" |
|
- "git+https://github.com/cloneofsimo/lora.git@799c17aef2a475641fb70d68a6992de4fc325ce4" |
|
- "opencv-python==4.7.0.68" |
|
- "torchvision==0.14.0" |
|
- "mediapipe==0.9.0.1" |
|
- "flake8==5.0.4" |
|
|
|
predict: "$PREDICT_FILE:Predictor" |
|
|
