Skip to content
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

[refactor] LoRA tests #9481

Merged
merged 5 commits into from
Sep 21, 2024
Merged

[refactor] LoRA tests #9481

merged 5 commits into from
Sep 21, 2024

Conversation

a-r-r-o-w
Copy link
Member

What does this PR do?

Minor cleanup of the lora tests

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.

@yiyixuxu @sayakpaul

@a-r-r-o-w a-r-r-o-w changed the title Lora tests refactor [refactor] LoRA tests Sep 20, 2024
Copy link
Member

@sayakpaul sayakpaul left a comment

Choose a reason for hiding this comment

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

Looks good, thanks!

Copy link
Collaborator

@yiyixuxu yiyixuxu left a comment

Choose a reason for hiding this comment

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

very nice clean up!


denoiser_to_checked = pipe.unet if self.unet_kwargs is not None else pipe.transformer
self.assertTrue(check_if_lora_correctly_set(denoiser_to_checked), "Lora not correctly set in Unet")
denoiser = pipe.transformer if self.unet_kwargs is None else pipe.unet
Copy link
Collaborator

Choose a reason for hiding this comment

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

very nice!

@sayakpaul sayakpaul merged commit e5d0a32 into main Sep 21, 2024
14 checks passed
@sayakpaul sayakpaul deleted the lora-tests-refactor branch September 21, 2024 01:40
@sayakpaul
Copy link
Member

Thanks for the refactor!

leisuzz pushed a commit to leisuzz/diffusers that referenced this pull request Oct 11, 2024
* refactor scheduler class usage

* reorder to make tests more readable

* remove pipeline specific checks and skip tests directly

* rewrite denoiser conditions cleaner

* bump tolerance for cog test
sayakpaul pushed a commit that referenced this pull request Dec 23, 2024
* refactor scheduler class usage

* reorder to make tests more readable

* remove pipeline specific checks and skip tests directly

* rewrite denoiser conditions cleaner

* bump tolerance for cog test
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