-
Notifications
You must be signed in to change notification settings - Fork 2
/
cog.yaml
40 lines (35 loc) · 1.41 KB
/
cog.yaml
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
# Configuration for Cog ⚙️
# Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md
image: "r8.im/laion-ai/deep-image-diffusion-prior"
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.8"
# a list of packages in the format <package-name>==<version>
python_packages:
- certifi==2022.6.15
- charset-normalizer==2.1.0
- einops==0.4.1
- idna==3.3
- madgrad==1.2
- numpy==1.23.0
- Pillow==9.2.0
- requests==2.28.1
- torch==1.10.1
- torchvision==0.11.2
- typing_extensions==4.3.0
- urllib3==1.26.9
# commands run after the environment is setup
run:
- mkdir -p /root/.cache/clip && wget -c --output-document "/root/.cache/clip/ViT-L-14.pt" "https://openaipublic.azureedge.net/clip/models/b8cca3fd41ae0c99ba7e8951adf17d267cdb84cd88be6f7c2e0eca1737a03836/ViT-L-14.pt"
- pip install 'git+https://github.com/openai/glide-text2im'
- pip install 'git+https://github.com/Veldrovive/DALLE2-pytorch@f4b687798d367fc434d8127ab31141f0fea0db26'
- pip install 'git+https://github.com/nousr/deep-image-prior.git'
# - pip install 'git+https://github.com/nousr/DALLE2-pytorch.git'
# predict.py defines how predictions are run on your model
predict: "predict.py:Predictor"