Add CogVideoX Diffusers checkpoint conversion - #14651
Add CogVideoX Diffusers checkpoint conversion#14651anubhavsingh23906-code wants to merge 1 commit into
Conversation
|
Hi @anubhavsingh23906-code, thanks for the PR! It does not appear to link an issue it fixes. If this PR addresses an existing issue, please add a closing keyword (e.g. Please note that PRs without a linked issue are likely to be automatically closed 10 days after this notice. Once the PR links an issue (or gets the |
|
Thanks for the reminder. I checked the existing Diffusers issues and could not find a tracked issue specifically This PR is intended to add the missing reverse conversion utility rather than fix a
|
What does this PR do?
Adds a new conversion script to convert CogVideoX transformer checkpoints from Diffusers format to the CogVideoX checkpoint format.
Changes
scripts/convert_diffusers_to_cogvideox.pyValidation
CogVideoXTransformer3DModelcan be instantiated successfully.--help.git diff --checkpasses.Testing
The conversion logic has currently been validated with synthetic state dictionaries. Real checkpoint conversion has not yet been performed because a suitable source checkpoint was not available locally.
Dependencies
No new dependencies were added.
Self-review