A port of the notebook in lucidrains/imagen-pytorch#24 to run locally using VS Code dev containers and WSL 2 with GPU support with some extra goodies.
- Ensure you have Docker Desktop and WSL2 installed and are running at least Windows 10 21H2 or Windows 11
- Open this repository in VSCode
- When prompted, reopen in container
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! 🙂