Downloads · 30 days
6
10% of all-time downloads
diffusionbreakout
kerzgrr/diffusionbreakout
diffusionbreakout 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.
An action-conditioned Breakout world model. Hold A / D (or the arrow keys) and the model generates the next frame.
All-time downloads
62
Public
Repo size
282 MB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.safetensors281 MB · 100%
At a glance
- Task
- other
- Library
- pytorch
- License
- apache-2.0
- Access
- Public
- Created
- Jul 18, 2026
- Updated
- Jul 18, 2026
- SHA
- 46695f2d
Model card
From the Hugging Face model README
Diffusion Breakout
An action-conditioned Breakout world model. Hold A / D (or the arrow keys) and the model generates the next frame.
128x128 at 6 FPS with 12 frames of latent history. The model has about 70M parameters and uses the frozen SDXL VAE from madebyollin/sdxl-vae-fp16-fix.
Run it
Requires a CUDA GPU with BF16 support.
pip install torch numpy pillow safetensors huggingface_hub diffusers
hf download kerzgrr/diffusionbreakout live_infer.py --local-dir .
python live_infer.py --steps 4
To keep the weights in a local folder:
hf download kerzgrr/diffusionbreakout \
--local-dir checkpoints/diffusionbreakout \
--include "ema.safetensors" \
--include "config.json" \
--include "live_infer.py"
python checkpoints/diffusionbreakout/live_infer.py \
--local-dir checkpoints/diffusionbreakout \
--steps 4 \
--window-scale 6
Click the window for focus. A / Left moves left; D / Right moves right.
Files
| file | contents |
|---|---|
ema.safetensors | action-conditioned EMA weights |
config.json | model, video, codec, and training settings |
live_infer.py | single-file live inference |
demo.gif | live rollout |
Actions are encoded as [left, right].
- Task
- other
- Library
- pytorch
- License
- apache-2.0
- Created
- Jul 18, 2026
- Updated
- Jul 18, 2026