This commit is contained in:
Ryan Dick 2023-09-14 23:49:02 -04:00
parent c104807201
commit 16664da5b6

View File

@ -511,7 +511,9 @@ class ControlNetFolderProbe(FolderProbeBase):
else ( else (
BaseModelType.StableDiffusion2 BaseModelType.StableDiffusion2
if dimension == 1024 if dimension == 1024
else BaseModelType.StableDiffusionXL if dimension == 2048 else None else BaseModelType.StableDiffusionXL
if dimension == 2048
else None
) )
) )
if not base_model: if not base_model: