imagen-notebook

A port of the notebook in https://github.com/lucidrains/imagen-pytorch/issues/24 to run locally using VS Code dev containers and WSL 2 with GPU support with some extra goodies.

How to use

1. Ensure you have Docker Desktop and WSL2 installed and are running at least Windows 10 21H2 or Windows 11
2. Open this repository in VSCode
3. When prompted, reopen in container

Credit

Many thanks to @Mimocro for creating the original colab notebook, @lucidrains for the imagen-pytorch implementation and the Google Research, Brain Team for imagen itself! 🙂