Repository navigation

Use torch.utils.checkpoint in mmdit forward loop when enabled to redu… - #1265

Merged

SWivid merged 1 commit into

SWivid:mainfrom

ZhikangNiu:main

Feb 14, 2026

Merged

Use torch.utils.checkpoint in mmdit forward loop when enabled to redu…#1265
SWivid merged 1 commit into
SWivid:mainfrom
ZhikangNiu:main

Commits on Feb 14, 2026
- Use torch.utils.checkpoint in mmdit forward loop when enabled to reduce memory usage.
  ZhikangNiu
  committed
  bb5526f