Add live Breakout demo GIF

Files changed (3) hide show

.gitattributes CHANGED Viewed

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text

 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+demo.gif filter=lfs diff=lfs merge=lfs -text

README.md CHANGED Viewed

@@ -15,7 +15,7 @@ tags:
 An action-conditioned Breakout world model. Hold **A / D** (or the arrow keys) and the model generates the next frame.
-![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`.
@@ -54,6 +54,6 @@ Click the window for focus. **A / Left** moves left; **D / Right** moves right.
 | `ema.safetensors` | action-conditioned EMA weights |
 | `config.json` | model, video, codec, and training settings |
 | `live_infer.py` | single-file live inference |
-| `preview.png` | validation strip |
 Actions are encoded as `[left, right]`.

 An action-conditioned Breakout world model. Hold **A / D** (or the arrow keys) and the model generates the next frame.
+![demo](demo.gif)
 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`.
 | `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]`.

demo.gif ADDED Viewed