Due to the nature of colab, installing these requirements in a cell requires a runtime restart to properly load all libraries and ensure that pixray runs smoothly. After installation, the cell for installation need not be run again for the rest of the session, even after more runtime restarts.
After restarting every time, this line should be run to ensure that the python interpreter knows where to find the pixray module.
import sys
sys.path.append("pixray")
a simple working snippet for Colab may be: (note that pixray also needs a models directory, at the moment unfortunately)
from IPython.utils import io
with io.capture_output() as captured:
!rm -Rf pixray
!git clone https://github.com/pixray/pixray
!pip install -r pixray/requirements.txt
!rm -Rf v-diffusion-pytorch
!git clone https://github.com/pixray/v-diffusion-pytorch
# not sure why this setup hack is necessary
!cp -r v-diffusion-pytorch/diffusion pixray/.
!pip uninstall -y tensorflow
!git clone https://github.com/pixray/diffvg
%cd diffvg
!git submodule update --init --recursive
!python setup.py install
%cd ..
!pip freeze | grep torch
import os
if not os.path.isfile("first_init_complete"):
!mkdir -p models
os.mknod("first_init_complete")