Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix inpaint model detection

Browse Source
This commit is contained in:
Lincoln Stein 2023-07-19 13:30:00 -04:00
parent 495df9fd1b
commit a1251c8e04
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
invokeai/backend/model_management/model_probe.py
View File

@ -39,6 +39,7 @@ class ModelProbe(object):
CLASS2TYPE = { CLASS2TYPE = {
'StableDiffusionPipeline' : ModelType.Main, 'StableDiffusionPipeline' : ModelType.Main,
'StableDiffusionInpaintPipeline' : ModelType.Main,
'StableDiffusionXLPipeline' : ModelType.Main, 'StableDiffusionXLPipeline' : ModelType.Main,
'StableDiffusionXLImg2ImgPipeline' : ModelType.Main, 'StableDiffusionXLImg2ImgPipeline' : ModelType.Main,
'AutoencoderKL' : ModelType.Vae, 'AutoencoderKL' : ModelType.Vae,
@ -401,7 +402,7 @@ class PipelineFolderProbe(FolderProbeBase):
in_channels = conf['in_channels'] in_channels = conf['in_channels']
if in_channels == 9: if in_channels == 9:
return ModelVariantType.Inpainting return ModelVariantType.Inpaint
elif in_channels == 5: elif in_channels == 5:
return ModelVariantType.Depth return ModelVariantType.Depth
elif in_channels == 4: elif in_channels == 4: