stable-diffusion-2D/cog.yaml at master · webaverse-studios/stable-diffusion-2D · GitHub

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
# Configuration for Cog ⚙️
# Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md

build:
  cuda: "11.2"
  # set to true if your model requires a GPU
  gpu: true

  # a list of ubuntu apt packages to install
  system_packages:
    - "sox"
    - "libsndfile1"
    - "ffmpeg"

  # python version in the form '3.8' or '3.8.12'
  python_version: "3.7.15"

  # a list of packages in the format <package-name>==<version>

  python_packages:
    - "git+https://github.com/takuma104/diffusers.git@9a37409663a53f775fa380db332d37d7ea75c915"
    - "numpy==1.21.6"
    - "openpyxl==3.0.10"
    - "PyOpenGL==3.1.6"
    - "python-dateutil==2.8.2"
    - "python-slugify==6.1.2"
    - "python-utils==3.4.5"
    - "Pillow==7.1.2"
    - "scikit-image==0.18.3"
    - "scikit-learn==1.0.2"
    - "scipy==1.7.3"
    - "torch==1.12.1"
    - "torchaudio==0.12.1"
    - "torchsummary==1.5.1"
    - "torchtext==0.13.1"
    - "torchvision==0.13.1"
    - "tornado==6.0.4"
    - "zipp==3.10.0"
    - "opencv-contrib-python==4.6.0.66"
    - "opencv-python==4.6.0.66"
    - "opencv-python-headless==4.6.0.66"

  run:
    - "pip install git+https://github.com/huggingface/transformers.git@fe9152f67c61c9af4721fdc9abbc9578acf5f16f"
    - "pip install openai"
# predict.py defines how predictions are run on your model
predict: "predict.py:Predictor"