Skip to content

[tests] Add torch.compile() test for WanTransformer3DModel #11451

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 2 commits into from

Conversation

tongyu0924
Copy link
Contributor

@tongyu0924 tongyu0924 commented Apr 29, 2025

What does this PR do?

This PR adds a torch.compile() test for the WanTransformer3DModel class.

Part of #11430

Before submitting

Who can review?

Anyone in the community is free to review the PR once the tests have passed. Feel free to tag
members/contributors who may be interested in your PR.

@tongyu0924 tongyu0924 closed this Apr 30, 2025
@sayakpaul
Copy link
Member

@tongyu0924 any reason why this was closed?

@tongyu0924
Copy link
Contributor Author

@tongyu0924 any reason why this was closed?

Hi! I closed it earlier since I wasn't really sure if it was right. Just reopened it—figure it's okay to leave it open now. Thanks for following up!

@tongyu0924 tongyu0924 reopened this Apr 30, 2025
@sayakpaul
Copy link
Member

Let us know if it's ready for a review.

@tongyu0924
Copy link
Contributor Author

Let us know if it's ready for a review.

Thanks! It's ready for review now. Let me know if anything needs adjustment

@sayakpaul
Copy link
Member

Closing in favor of #11475 and that PR already includes your contribution:

image

@sayakpaul sayakpaul closed this May 1, 2025
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.

2 participants