File size: 2,155 Bytes
1976a91 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 |
# Configuration for Cog ⚙️
# https://replicate.com/docs/guides/push-a-model
# prerequisite：https://docs.docker.com/engine/install/ubuntu/#set-up-the-repository `dockerd` to start docker
# Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md
# !!!! recommend 60G disk space for cog docker
build:
# set to true if your model requires a GPU
gpu: true
# a list of ubuntu apt packages to install
system_packages:
- "libgl1-mesa-glx"
- "libglib2.0-0"
# python version in the form '3.8' or '3.8.12'
python_version: "3.10.4"
# a list of packages in the format <package-name>==<version>
python_packages:
- blendmodes==2022
- transformers==4.25.1
- accelerate==0.12.0
- basicsr==1.4.2
- gfpgan==1.3.8
- gradio==3.16.2
- numpy==1.23.3
- Pillow==9.4.0
- realesrgan==0.3.0
# - torch==1.13.1+cu117
# - --extra-index-url https://download.pytorch.org/whl/cu117
# - torchvision==0.14.1+cu117
# - --extra-index-url https://download.pytorch.org/whl/cu117
- omegaconf==2.2.3
- pytorch_lightning==1.7.6
- scikit-image==0.19.2
- fonts
- font-roboto
- timm==0.6.7
- piexif==1.1.3
- einops==0.4.1
- jsonmerge==1.8.0
- clean-fid==0.1.29
- resize-right==0.0.2
- torchdiffeq==0.2.3
- kornia==0.6.7
- lark==1.1.2
- inflection==0.5.1
- GitPython==3.1.27
- torchsde==0.2.5
- safetensors==0.2.7
- httpcore<=0.15
- fastapi==0.90.1
# - open_clip_torch
- git+https://github.com/mlfoundations/open_clip.git@bb6e834e9c70d9c27d0dc3ecedeebeaeb1ffad6b
- git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1
# commands run after the environment is setup
run:
- "pip3 install torch==1.13.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117"
- "pip3 install torchvision==0.14.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117"
- "echo env is ready!"
# https://replicate.com/wolverinn/chill_watcher
image: "r8.im/wolverinn/chill_watcher"
# predict.py defines how predictions are run on your model
predict: "predict.py:Predictor"
|