pixray/cog.yaml at master · LightricksResearch/pixray · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
build:
  gpu: true
  cuda: "10.2"
  python_version: "3.8"
  python_requirements: "requirements.txt"
  run:
    - "apt-get update && apt install -y libgl1-mesa-glx"
    # CMAKE_PREFIX_PATH fix is from https://github.com/readthedocs/readthedocs.org/issues/5867
    - "git clone https://github.com/pixray/diffvg && cd diffvg && git submodule update --init --recursive && CMAKE_PREFIX_PATH=$(pyenv prefix) DIFFVG_CUDA=1 python setup.py install && pip install basicsr"
# predict: "cogrun.py:Text2Image"
# predict: "cogrun.py:Text2Pixel"
# predict: "cog_genesis.py:GenesisPredictor"
# predict: "cogrun.py:PixrayApi"
# predict: "cogrun.py:Tiler"
# predict: "cogrun.py:PixrayRaw"
predict: "cogrun.py:PixrayVdiff"
# OLDSTUFF
# predict: "cogrun.py:BasePixrayPredictor"
# predict: "cogrun.py:PixrayVqgan"
# predict: "cogrun.py:PixrayPixel"