Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
51b0184969 |
No files matched your search
@@ -278,6 +278,20 @@ def tiny_sd3_builder() -> str:
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def tiny_sdxl_builder() -> str:
|
||||||
|
# The UNet's architecture (block_out_channels, transformer_layers_per_block, etc.)
|
||||||
|
# is hardcoded in SDXLUNet2DConditionModel rather than read from config, so it
|
||||||
|
# can't be shrunk here; only the two CLIP text encoders are reduced.
|
||||||
|
return build_tiny_from_configs(
|
||||||
|
"StableDiffusionXLPipeline",
|
||||||
|
"stabilityai/stable-diffusion-xl-base-1.0",
|
||||||
|
transform={
|
||||||
|
"text_encoder": _shrink_flux_clip_text_encoder,
|
||||||
|
"text_encoder_2": _shrink_flux_clip_text_encoder,
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def tiny_flux2_builder() -> str:
|
def tiny_flux2_builder() -> str:
|
||||||
def shrink_text_encoder(config: dict) -> dict:
|
def shrink_text_encoder(config: dict) -> dict:
|
||||||
config["tie_word_embeddings"] = False
|
config["tie_word_embeddings"] = False
|
||||||
|
|||||||
@@ -181,4 +181,10 @@ DIFFUSION_TEST_SETTINGS = {
|
|||||||
supported_tasks=[DiffusionTasks.TEXT_TO_IMAGE],
|
supported_tasks=[DiffusionTasks.TEXT_TO_IMAGE],
|
||||||
model_type_marker=ModelTypeMarker.DIFFUSION,
|
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]],
|
||||||
|
),
|
||||||
}
|
}
|
||||||
@@ -84,7 +84,6 @@ EXCLUDED_MODELS = [
|
|||||||
"HiDreamImagePipeline",
|
"HiDreamImagePipeline",
|
||||||
"HiDreamO1ImagePipeline",
|
"HiDreamO1ImagePipeline",
|
||||||
"DreamZeroPipeline",
|
"DreamZeroPipeline",
|
||||||
"StableDiffusionXLPipeline",
|
|
||||||
"Gr00tN1d7Pipeline",
|
"Gr00tN1d7Pipeline",
|
||||||
"Pi0Pipeline",
|
"Pi0Pipeline",
|
||||||
"Pi05Pipeline",
|
"Pi05Pipeline",
|
||||||
|
|||||||
Reference in new issue
Block a user