Skip to content

test_load_wan

GPU loading + forward smoke test for WanModel.

Loads the real Wan2.1 1.3B checkpoint via WanModel.__init__ and runs one transformer forward pass on synthetic inputs. Catches loader or forward-signature regressions in fastvideo.train.models.wan.WanModel and the underlying WanTransformer3DModel.

Classes

Functions