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 (
BaseModelType.StableDiffusion2
if dimension == 1024
else BaseModelType.StableDiffusionXL if dimension == 2048 else None
else BaseModelType.StableDiffusionXL
if dimension == 2048
else None
)
)
if not base_model: