Google Colab

[woctezuma]

For info for others, to run the code on Google Colab (with GPU toggled ON beforehand), you only have to run this cell:

%pip install -q requests tqdm ftfy

%cd /content
!git clone --recursive https://github.com/crowsonkb/v-diffusion-pytorch.git

%cd /content/v-diffusion-pytorch
%mkdir -p checkpoints
!curl https://v-diffusion.s3.us-west-2.amazonaws.com/cc12m_1_cfg.pth -o checkpoints/cc12m_1_cfg.pth

!./cfg_sample.py "the rise of consciousness":5 -n 4 -bs 4 --seed 0

The README is only missing the installation of ftfy.

With a Tesla K80, the expected run-time seems to be ~ 40 minutes. I have not tried till the end.