Skip to content

refactor unet_2d_condition tests#13902

Closed
akshan-main wants to merge 1 commit into
huggingface:mainfrom
akshan-main:tests-refactor-unet-2d-condition
Closed

refactor unet_2d_condition tests#13902
akshan-main wants to merge 1 commit into
huggingface:mainfrom
akshan-main:tests-refactor-unet-2d-condition

Conversation

@akshan-main

Copy link
Copy Markdown
Contributor

What does this PR do?

Part of the ongoing modeling-test migration (following #13369 and #13153). Migrates the UNet2DConditionModel test suite to the mixin-based structure (Config + ModelTesterMixin / TrainingTesterMixin / MemoryTesterMixin / AttentionTesterMixin / TorchCompileTesterMixin / LoraHotSwappingForModelTesterMixin). Last of the UNet migrations.

The model-specific tests (IP-adapter, custom diffusion, projection variants, sharded checkpoints, attn-procs warnings) are kept, the deprecated xformers and attention-slicing tests are dropped, and the @slow integration class is unchanged.

Before submitting

Who can review?

@sayakpaul @DN6

@github-actions github-actions Bot added tests size/L PR with diff > 200 LOC labels Jun 9, 2026
@sayakpaul

Copy link
Copy Markdown
Member

This one is here: #13153

@akshan-main

Copy link
Copy Markdown
Contributor Author

Sounds good, I'll defer unet_2d_condition to #13153 since you've got a thorough version there already, closing this one.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

size/L PR with diff > 200 LOC tests

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants