← 返回架构图
❄️

DiT Low-Noise Model

wan/modules/model.py · WanModel · t < boundary
低噪阶段
当 timestep t < boundary 时激活,接管高噪模型的输出作为输入。 专注于细节精修:纹理、边缘锐化、高频细节生成。 与高噪模型共享相同的 WanModel 架构,但权重完全独立(分别从不同 checkpoint 加载)。
双模型切换逻辑
boundaryconfig.boundary × num_train_timesteps
高噪激活t ≥ boundary → high_noise_model.to(cuda)
低噪激活t < boundary → low_noise_model.to(cuda)
offload非激活模型卸载到 CPU 释放显存
Fast 变体差异 (WanModelFast)
KV Cacheself_kv_cache 跨块缓存 K/V
因果注意力sp_attn_forward_causal:只看过去帧
逐块推理chunk-by-chunk,每次处理 chunk_size 帧
prewarm首块全量计算填充 KV Cache