Skip to content

Conversation

sayakpaul
Copy link
Member

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.

@sayakpaul sayakpaul requested a review from DN6 April 5, 2024 05:00
Comment on lines -374 to -378
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)
Copy link
Member Author

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.

@sayakpaul
Copy link
Member Author

@DN6 a gentle ping here.

Copy link
Collaborator

@DN6 DN6 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM 👍🏽

@yiyixuxu yiyixuxu merged commit b2323aa into main Apr 10, 2024
@sayakpaul sayakpaul deleted the tests/reduce-block-sizes-in-pipelines branch April 11, 2024 03:18
sayakpaul added a commit that referenced this pull request Dec 23, 2024
* give it a shot.

* print.

* correct assertion.

* gather results from the rest of the tests.

* change the assertion values where needed.

* remove print statements.
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.

3 participants