-
Notifications
You must be signed in to change notification settings - Fork 6.3k
[Tests] reduce the model sizes in the SD fast tests #7580
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
Conversation
def test_ip_adapter_single(self): | ||
expected_pipe_slice = None | ||
if torch_device == "cpu": | ||
expected_pipe_slice = np.array([0.3203, 0.4555, 0.4711, 0.3505, 0.3973, 0.4650, 0.5137, 0.3392, 0.4045]) | ||
return super().test_ip_adapter_single(expected_pipe_slice=expected_pipe_slice) |
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.
No point in keeping this test this way because we are reducing the model sizes dramatically so this won't contribute too much. LMK if that doesn't make sense.
@DN6 a gentle ping here. |
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.
LGTM 👍🏽
* give it a shot. * print. * correct assertion. * gather results from the rest of the tests. * change the assertion values where needed. * remove print statements.
What does this PR do?
Reduces the model sizes in the Stable Diffusion fast tests. My plan is to have this PR as a reference and open the rest of the tests that would benefit from this to the community. Hope that makes sense.
Cc: @yiyixuxu for visibility.