We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
from_single_file
1 parent cf2c49b commit 214990eCopy full SHA for 214990e
src/diffusers/loaders/single_file_utils.py
@@ -449,6 +449,8 @@ def infer_diffusers_model_type(checkpoint):
449
):
450
if CHECKPOINT_KEY_NAMES["v2"] in checkpoint and checkpoint[CHECKPOINT_KEY_NAMES["v2"]].shape[-1] == 1024:
451
model_type = "inpainting_v2"
452
+ elif CHECKPOINT_KEY_NAMES["xl_base"] in checkpoint:
453
+ model_type = "xl_inpaint"
454
else:
455
model_type = "inpainting"
456
0 commit comments