shengshu-ai/minWM
A Minimal and Elegant Framework & Tutorial for Real-Time Interactive World Models
What it solves
minWM is a full-stack framework designed to help researchers and developers transition from a standard text-to-video (T2V) foundation model to a real-time, action-conditioned video world model. It removes the need for users to reverse-engineer complex pipelines by providing a complete, open-source path from data preparation to training and inference.
How it works
The framework implements a four-stage pipeline to transform bidirectional diffusion models into 4-step autoregressive (AR) student models for real-time performance:
- Data Preparation: Processes raw videos and camera trajectories (poses) into training-ready latents.
- Bidirectional SFT: Initial supervised fine-tuning of the foundation model.
- Distillation: A multi-stage process involving Teacher Forcing AR diffusion, Causal ODE/CD distillation, and Asymmetric DMD with self-rollout.
- Inference: Uses 4-step DMD inference with support for camera-trajectory control via pose strings (e.g., "a4,w8") or JSON files.
It supports multiple backbones, including Wan 2.1 and HunyuanVideo 1.5, using a unified trainer and dataset abstraction.
Who it’s for
It is primarily aimed at graduate students, independent researchers, and junior labs entering the world-model space who want a structured, runnable framework rather than a standalone model.
Highlights
- Full-Stack Pipeline: Open-sources the entire data $\rightarrow$ training $\rightarrow$ inference flow.
- Multi-Backbone Support: Compatible with both Wan 2.1 and HunyuanVideo 1.5 architectures.
- LLM-Assisted Debugging: Includes "Claude Skills" to help users diagnose training failures (like loss NaN or camera drift) and integrate new backbones.
- Real-Time Performance: Distills models down to 4-step inference for interactive video generation.
Related
- Project
- Project
- Project
- Project
- Project