refactor bria pipeline tests to the new mixin structure - #14547
Conversation
|
/diffusers-bot pytest tests/pipelines/bria/test_pipeline_bria.py |
|
|
|
@akshan-main additionally, let's also add the |
|
/diffusers-bot pytest tests/pipelines/bria/test_pipeline_bria.py |
|
✅ |
|
Added. Note BriaPipeline doesn't subclass LoraBaseMixin, so the class skips through the mixin's setup gate for now. |
|
Sorry. I should have been clear here. If the corresponding pipeline doesn't already have a separate LoRA tester under tests/lora, then no need to add. This means we can remove the LoRA bits you just added. I am sorry for the confusion. |
|
Hi @akshan-main, thanks for the PR! It does not appear to link an issue it fixes. If this PR addresses an existing issue, please add a closing keyword (e.g. |
|
No worries, removed. |
What does this PR do?
Refactors the Bria pipeline tests to the new mixin structure from #14113: a
BriaPipelineTesterConfigplusTestBriaPipeline(PipelineTesterMixin)andTestBriaPipelineMemory(MemoryTesterMixin), with the slow class ported to pytest asTestBriaPipelineSlow.Existing coverage is kept:
test_bria_different_promptsand both output-shape tests, with dummy inputs switched tooutput_type="pt"and a fresh CPU slice intest_inference. The old silenttest_encode_prompt_works_in_isolationoverride is now an explicit skip with the diagnosed reason (BriaPipeline.__init__dereferences the vae config, so it cannot be built with text-only components). The float16/dtype overrides are dropped in favor of the framework versions, which pass unmodified.Before submitting
Who can review?
@sayakpaul