KEMBAR78
[refactor] LoRA tests by a-r-r-o-w · Pull Request #9481 · huggingface/diffusers · GitHub
Skip to content

Conversation

a-r-r-o-w
Copy link
Contributor

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