[Tests] Add tiny model test for StableDiffusionXLPipeline
Covers base text-to-image plus CPU offload; UNet stays full-size since its architecture is hardcoded rather than config-driven. Co-Authored-By: Claude <noreply@anthropic.com> Signed-off-by: Nick Cao <ncao@redhat.com>
This commit is contained in:
1 parent
9146284c16
commit
51b0184969
3 files changed
+20
-1
No files matched your search
@@ -181,4 +181,10 @@ DIFFUSION_TEST_SETTINGS = {
|
||||
supported_tasks=[DiffusionTasks.TEXT_TO_IMAGE],
|
||||
model_type_marker=ModelTypeMarker.DIFFUSION,
|
||||
),
|
||||
"StableDiffusionXLPipeline": DiffusionModelTestOpts(
|
||||
model="stabilityai/stable-diffusion-xl-base-1.0",
|
||||
builder=diff_model_builders.tiny_sdxl_builder,
|
||||
supported_tasks=[DiffusionTasks.TEXT_TO_IMAGE],
|
||||
extra_test_groups=[[DiffusionAccs.CPU_OFFLOAD]],
|
||||
),
|
||||
}
|
||||
Reference in new issue
Block a user