-
Notifications
You must be signed in to change notification settings - Fork 4.8k
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
[Tests] reduce the model size in the dit test #7898
base: main
Are you sure you want to change the base?
Conversation
…-2001/diffusers into dance_diffusion_test_fast
…-2001/diffusers into dance_diffusion_test_fast
Hi @sayakpaul, can you pls review. |
Why are you deleting |
Hi sorry, I thought I added the |
Hi @sayakpaul, can you pls review this once? Thanks |
tests/pipelines/dit/test_dit.py
Outdated
num_attention_heads=2, | ||
in_channels=4, | ||
out_channels=8, | ||
out_channels=4, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This needs to be 8 as that is how guidance is formulated in DiT.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Noted and updated.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks. Left a comment.
Hi @sayakpaul, can you pls review this. |
What does this PR do?
Reduces the model sizes in the DIT tests.
Fixes #7677
Before submitting
documentation guidelines, and
here are tips on formatting docstrings.
Who can review?
@sayakpaul