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

[Flux] add lora integration tests. #9353

Merged
merged 5 commits into from
Sep 19, 2024
Merged

Conversation

sayakpaul
Copy link
Member

Added integration tests for three popular LoRA trainers for Flux:

  • Xlabs
  • Last Ben
  • Kohya

@sayakpaul sayakpaul requested review from DN6 and yiyixuxu September 3, 2024 07:34
@sayakpaul
Copy link
Member Author

@yiyixuxu a gentle ping here.

@slow
@require_torch_gpu
@require_peft_backend
class FluxLoRAIntegrationTests(unittest.TestCase):
Copy link
Member Author

Choose a reason for hiding this comment

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

Can add a skip tag like we have for the Flux pipeline tests:

@unittest.skip("We cannot run inference on this model with the current CI hardware")

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.

thank you!

@sayakpaul sayakpaul merged commit d13b0d6 into main Sep 19, 2024
14 checks passed
@sayakpaul sayakpaul deleted the lora-flux-integration-tests branch September 19, 2024 03:51
leisuzz pushed a commit to leisuzz/diffusers that referenced this pull request Oct 11, 2024
* add lora integration tests.

* internal note

* add a skip marker.
sayakpaul added a commit that referenced this pull request Dec 23, 2024
* add lora integration tests.

* internal note

* add a skip marker.
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.

2 participants