Downloads · 30 days
8
7% of all-time downloads
diffusionpong
kerzgrr/diffusionpong
diffusionpong is a other model from kerzgrr. Use it for the other task on the model card, and read the license before you ship it in a product. It is set up for pytorch. The card lists the license as apache-2.0.
A small action-conditioned world model that learned to keep a Pong match going. You hold W / S, it invents the next frame.
All-time downloads
111
Public
Repo size
471 MB
Likes
2
Public
Repo makeup
Click a slice to open those files.
.safetensors235 MB · 100%
At a glance
- Task
- other
- Library
- pytorch
- License
- apache-2.0
- Access
- Public
- Created
- Jul 17, 2026
- Updated
- Jul 17, 2026
- SHA
- f6d8d2a6
Model card
From the Hugging Face model README
Diffusion Pong
A small action-conditioned world model that learned to keep a Pong match going. You hold W / S, it invents the next frame.
128x128 at 6 FPS with a 12-frame latent history. Codec is a frozen SDXL VAE (madebyollin/sdxl-vae-fp16-fix). Needs a CUDA GPU with BF16.
Play it
pip install torch numpy pillow safetensors huggingface_hub diffusers
hf download kerzgrr/diffusionpong live_infer.py --local-dir .
python live_infer.py --steps 2
Or pull weights into a folder yourself:
hf download kerzgrr/diffusionpong --local-dir checkpoints/diffusionpong --include "ema.safetensors" --include "config.json" --include "live_infer.py"
python checkpoints/diffusionpong/live_infer.py --local-dir checkpoints/diffusionpong --steps 2 --window-scale 7 --seed 42
Controls: W up, S down (A/D also work). Click the window once so it has focus. Clicking the canvas drops a yellow ball into the latent history for a few frames if you want to poke it.
What's in the files
| file | what |
|---|---|
ema.safetensors | playable weights |
config.json | video / model / codec settings used at train time |
live_infer.py | live play script |
demo.gif | live rollout clip |
Action vector is [W, S] as floats in {0,1}. Hold both and it treats that as neutral.
If something breaks, it's probably the VAE download or a GPU that can't do BF16.
- Task
- other
- Library
- pytorch
- License
- apache-2.0
- Created
- Jul 17, 2026
- Updated
- Jul 17, 2026