1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
This commit is contained in:
Dhruv Nair
2023-12-28 12:40:31 +00:00
parent 2616e03062
commit 7db4f50e85

View File

@@ -219,7 +219,7 @@ def build_additional_components(
)
return paint_by_example_components
if pipeline_class_name == "StableDiffusionXLImg2ImgPipeline":
if pipeline_class_name in ["StableDiffusionXLImg2ImgPipeline", "StableDiffusionXLInpaintPipeline"]:
model_type = infer_model_type(pipeline_class_name, original_config)
is_refiner = model_type == "SDXL-Refiner"
return {