Unverified Commit 214990e5 authored by Vishnu V Jaddipal's avatar Vishnu V Jaddipal Committed by GitHub
Browse files

Fix ```from_single_file``` for xl_inpaint (#9054)

parent cf2c49b1
...@@ -449,6 +449,8 @@ def infer_diffusers_model_type(checkpoint): ...@@ -449,6 +449,8 @@ def infer_diffusers_model_type(checkpoint):
): ):
if CHECKPOINT_KEY_NAMES["v2"] in checkpoint and checkpoint[CHECKPOINT_KEY_NAMES["v2"]].shape[-1] == 1024: if CHECKPOINT_KEY_NAMES["v2"] in checkpoint and checkpoint[CHECKPOINT_KEY_NAMES["v2"]].shape[-1] == 1024:
model_type = "inpainting_v2" model_type = "inpainting_v2"
elif CHECKPOINT_KEY_NAMES["xl_base"] in checkpoint:
model_type = "xl_inpaint"
else: else:
model_type = "inpainting" model_type = "inpainting"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment