bypass validation scheduler setup for special models #2539
Merged
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This pull request introduces a new mechanism for models to indicate if they require special handling when setting up schedulers during validation. Several model classes now override a new method to signal that scheduler setup should be delegated to the model pipeline, and the validation logic is updated to respect this flag.
Scheduler setup improvements:
requires_special_scheduler_setupto the base model class (common.py), allowing models to specify if they need custom scheduler initialization.validation.pyto checkrequires_special_scheduler_setupand skip default scheduler setup if the method returnsTrue.Model-specific overrides:
requires_special_scheduler_setupto returnTruefor the following models, indicating that they handle scheduler setup internally:ltxvideo(model.py)ltxvideo2(model.py)Sana(model.py)SanaVideo(model.py)Wan(model.py)