Skip to content

Commit 214990e

Browse files
authored
Fix from_single_file for xl_inpaint (#9054)
1 parent cf2c49b commit 214990e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/diffusers/loaders/single_file_utils.py

+2
Original file line numberDiff line numberDiff line change
@@ -449,6 +449,8 @@ def infer_diffusers_model_type(checkpoint):
449449
):
450450
if CHECKPOINT_KEY_NAMES["v2"] in checkpoint and checkpoint[CHECKPOINT_KEY_NAMES["v2"]].shape[-1] == 1024:
451451
model_type = "inpainting_v2"
452+
elif CHECKPOINT_KEY_NAMES["xl_base"] in checkpoint:
453+
model_type = "xl_inpaint"
452454
else:
453455
model_type = "inpainting"
454456

0 commit comments

Comments
 (0)