Google Colab
woctezuma opened this issue · 0 comments
woctezuma commented
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.