search

lucidrains / deep-daze

Simple command line tool for text to image generation using OpenAI's CLIP and Siren (Implicit neural representation network). Technique was originally created by https://twitter.com/advadnoun
MIT License
4.37k stars 327 forks source link

Colab TPU Runtime #142

Open metaphorz opened 3 years ago

metaphorz commented 3 years ago

I tried setting Colab Pro runtime to "TPU" but got errors. Is there a simple edit I can do in colab to make it work with the TPU architecture? I ask because I've spent all afternoon having a great time with deep daze and big sleep. They are fantastic! Maybe, the TPU will be faster--something I'd like to try. Thank you for this.

callisto-cloud commented 2 years ago

It's not working because Pytorch ist not working with TPU's as of today. However there is a branch of Pytorch which currently works with Google colab TPU but I guess you need to create branch of this repo and implement it yourself until someone else does it

Pytorch working with TPU