Skip to content

refactor unet_2d tests#13901

Merged
sayakpaul merged 3 commits into
huggingface:mainfrom
akshan-main:tests-refactor-unet-2d
Jun 10, 2026
Merged

refactor unet_2d tests#13901
sayakpaul merged 3 commits into
huggingface:mainfrom
akshan-main:tests-refactor-unet-2d

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 UNet2DModel test suites (the standard UNet, the LDM config, and NCSNpp) to the mixin-based structure (Config + ModelTesterMixin / TrainingTesterMixin / MemoryTesterMixin).

NCSNpp doesn't support layerwise casting, so those tests are skipped as before. The hub/pretrained integration tests are kept.

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 sayakpaul merged commit f809080 into huggingface:main Jun 10, 2026
13 checks passed
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