Enhance DiT and Wan mock data modules by initializing training data l…#65

Merged

abhinavg4 merged 3 commits intomainfrom

Nov 20, 2025

Commits on Nov 20, 2025

Enhance DiT and Wan mock data modules by initializing training data loaders as iterators. Update test configuration for DiT pretraining to include checkpoint save and load paths.
abhinavg4
committed
Remove unnecessary DataLoader assertion from Wan mock datamodule tests to streamline test execution.
abhinavg4
committed
Merge branch 'main' into dit_fix
abhinavg4
authored