* alias
* add tests
* feedback
* Fix failing test_dtype_alias pipeline tests
Cosmos pipelines keep the safety_checker in float32 and require it to be
passed explicitly on reload, mirroring their test_torch_dtype_dict
overrides. PRX skips dtype-dict/save-load tests because the custom
T5GemmaEncoder is not compatible with transformers v5; do the same for
test_dtype_alias.
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
---------
Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com>
* implement base model output caching in model-level tests
* single quotes
* memory
---------
Co-authored-by: dg845 <58458699+dg845@users.noreply.github.com>
* port final set of model tests and others
* fix extracter.
* fix cuda tests for models.
* Revert "fix extracter."
This reverts commit 178c4cb99b.
* Revert "port final set of model tests and others"
This reverts commit a92c70c081.
* fix more
* address reviewer feedback
---------
Co-authored-by: dg845 <58458699+dg845@users.noreply.github.com>
* refactor unet2d condition model tests.
* fix tests
* up
* fix
* Revert "fix"
This reverts commit 46d44b73d8.
* up
* recompile limit
* [tests] refactor test_models_unet_1d.py to use modular testing mixins
Refactor UNet1D model tests to follow the modern testing pattern using
BaseModelTesterConfig and focused mixin classes (ModelTesterMixin,
MemoryTesterMixin, TrainingTesterMixin, LoraTesterMixin).
Both UNet1D standard and RL variants now have separate config classes
and dedicated test classes organized by concern (core, memory, training,
LoRA, hub loading).
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
* [tests] refactor test_models_unet_2d.py to use modular testing mixins
Refactor UNet2D model tests (standard, LDM, NCSN++) to follow the
modern testing pattern. Each variant gets its own config class and
dedicated test classes organized by concern (core, memory, training,
LoRA, hub loading).
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
* [tests] refactor test_models_unet_3d_condition.py to use modular testing mixins
Refactor UNet3DConditionModel tests to follow the modern testing pattern
with separate classes for core, attention, memory, training, and LoRA.
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
* [tests] refactor test_models_unet_controlnetxs.py to use modular testing mixins
Refactor UNetControlNetXSModel tests to follow the modern testing
pattern with separate classes for core, memory, training, and LoRA.
Specialized tests (from_unet, freeze_unet, forward_no_control,
time_embedding_mixing) remain in the core test class.
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
* [tests] refactor test_models_unet_spatiotemporal.py to use modular testing mixins
Refactored the spatiotemporal UNet test file to follow the modern modular testing
pattern with BaseModelTesterConfig and focused test classes:
- UNetSpatioTemporalTesterConfig: Base configuration with model setup
- TestUNetSpatioTemporal: Core model tests (ModelTesterMixin, UNetTesterMixin)
- TestUNetSpatioTemporalAttention: Attention-related tests (AttentionTesterMixin)
- TestUNetSpatioTemporalMemory: Memory/offloading tests (MemoryTesterMixin)
- TestUNetSpatioTemporalTraining: Training tests (TrainingTesterMixin)
- TestUNetSpatioTemporalLoRA: LoRA adapter tests (LoraTesterMixin)
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
* remove test suites that are passed.
* fix consistencydecodervae tests
* Revert "fix consistencydecodervae tests"
This reverts commit 41b036b989.
---------
Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
Co-authored-by: dg845 <58458699+dg845@users.noreply.github.com>
