Spaces:
Paused
Paused
| # Configuration for Cog ⚙️ | |
| # Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md | |
| build: | |
| # set to true if your model requires a GPU | |
| gpu: true | |
| cuda: "12.1" | |
| # a list of ubuntu apt packages to install | |
| system_packages: | |
| - "libgl1-mesa-glx" | |
| - "libglib2.0-0" | |
| # python version in the form '3.11' or '3.11.4' | |
| python_version: "3.11" | |
| # a list of packages in the format <package-name>==<version> | |
| python_packages: | |
| - "opencv-python==4.9.0.80" | |
| - "transformers==4.37.0" | |
| - "accelerate==0.26.1" | |
| - "insightface==0.7.3" | |
| - "diffusers==0.25.1" | |
| - "onnxruntime==1.16.3" | |
| - "omegaconf==2.3.0" | |
| - "gradio==3.50.2" | |
| - "peft==0.8.2" | |
| - "transformers==4.37.0" | |
| - "controlnet-aux==0.0.7" | |
| # fix for pydantic issues in cog | |
| # https://github.com/replicate/cog/issues/1623 | |
| - albumentations==1.4.3 | |
| # commands run after the environment is setup | |
| run: | |
| - curl -o /usr/local/bin/pget -L "https://github.com/replicate/pget/releases/download/v0.6.0/pget_linux_x86_64" && chmod +x /usr/local/bin/pget | |
| # predict.py defines how predictions are run on your model | |
| predict: "cog/predict.py:Predictor" | |