Spaces:
Running
on
Zero
Running
on
Zero
| # Configuration for Cog ⚙️ | |
| # Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md | |
| build: | |
| gpu: true | |
| python_version: "3.8" | |
| python_packages: | |
| - "accelerate==0.25.0" | |
| - "diffusers==0.21.0" | |
| - "torch==2.0.1" | |
| - "pytorch_lightning==2.1.3" | |
| - "transformers==4.25.0" | |
| - "xformers" | |
| - "loralib==0.1.2" | |
| - "fairscale==0.4.13" | |
| - "opencv-python==4.9.0.80" | |
| - "chardet==5.2.0" | |
| - "einops==0.7.0" | |
| - "scipy==1.10.1" | |
| - "timm==0.9.12" | |
| run: | |
| - curl -o /usr/local/bin/pget -L "https://github.com/replicate/pget/releases/download/v0.3.1/pget" && chmod +x /usr/local/bin/pget | |
| # predict.py defines how predictions are run on your model | |
| predict: "predict.py:Predictor" | |