---
license: apache-2.0
library_name: pytorch
pipeline_tag: other
tags:
  - world-model
  - diffusion
  - breakout
  - video
  - causal
---

# Diffusion Breakout Base

The unconditional Breakout world model used as the starting point for the playable version.

![validation preview](preview.png)

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.

```bash
pip install torch numpy pillow safetensors huggingface_hub diffusers

hf download kerzgrr/diffusionbreakout-base live_infer.py --local-dir .
python live_infer.py --steps 1
```

To download everything first:

```bash
hf download kerzgrr/diffusionbreakout-base \
  --local-dir checkpoints/diffusionbreakout-base \
  --include "ema.safetensors" \
  --include "config.json" \
  --include "live_infer.py"

python checkpoints/diffusionbreakout-base/live_infer.py \
  --local-dir checkpoints/diffusionbreakout-base \
  --steps 1 \
  --window-scale 6
```

This model is unconditional, so there are no paddle controls. It rolls forward from a fresh game context.

## Files

| file | contents |
|---|---|
| `ema.safetensors` | unconditional EMA weights |
| `config.json` | model, video, codec, and training settings |
| `live_infer.py` | single-file live inference |
| `preview.png` | validation strip |