Skip to content

Conversation

@abhinavg4
Copy link
Contributor

…oaders as iterators. Update test configuration for DiT pretraining to include checkpoint save and load paths.

…oaders as iterators. Update test configuration for DiT pretraining to include checkpoint save and load paths.
@abhinavg4
Copy link
Contributor Author

/ok to test 0e92599

@abhinavg4 abhinavg4 merged commit 47bdad5 into main Nov 20, 2025
16 checks passed
sajadn pushed a commit that referenced this pull request Nov 20, 2025
#65)

* 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.

* Remove unnecessary DataLoader assertion from Wan mock datamodule tests to streamline test execution.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants