---
library_name: pytorch
tags:
- video-generation
- world-model
- diffusion
- diffusion-forcing
datasets:
- TeaPearce/CounterStrike_Deathmatch
---

# NanoWM-L/2 · csgo · NanoWM-L/2 on CSGO

NanoWM-L/2 on CSGO (50k steps)

## Run identity
- **wandb**: https://wandb.ai/better_guidance/nano-world-model-phase2/runs/cosr7ivu
- **launcher**: `src/scripts/train/main/csgo.sh`
- **collection**: https://huggingface.co/collections/knightnemo/nano-world-model

## Training setup
| Key | Value |
|---|---|
| Architecture | NanoWM-L/2 |
| Dataset | csgo |
| Prediction | v |
| Noise schedule | squaredcos_cap_v2 (ZTSNR=True) |
| Steps | 50000 |
| Batch | 6/GPU |

## Loading

```python
import sys
from omegaconf import OmegaConf
from safetensors.torch import load_file
sys.path.insert(0, "src")
from models import get_models

cfg = OmegaConf.load("ckpt/config.yaml")
cfg.experiment.infra.compile = False
model = get_models(cfg).eval()
state_dict = load_file("ckpt/model.safetensors")
model.load_state_dict(state_dict, strict=True)
```