It is an educational repo demonstrating how to build a real-time Snake game using a Diffusion model. It was inspired by several great papers:
The goal was to create a similar implementation using Snake game due its simple logic. It took near 2 months of different experiments to get a ready-to-play model.
After several experiments, I selected the EDM diffusion model for its high performance with small sample steps. DDIM requires much more steps to achieve comparable quality.