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(mm): fix bug in control adapter probe default settings

Browse Source 
Wasn't checking for matches correctly.
This commit is contained in:
psychedelicious 2024-03-08 20:39:13 +11:00 committed by Brandon
parent a3570901f7
commit a10dccdd43
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
invokeai/backend/model_manager/probe.py
View File

@ -346,23 +346,25 @@ class ControlAdapterProbe(ProbeBase):
"mlsd": "mlsd_image_processor", "mlsd": "mlsd_image_processor",
"depth": "depth_anything_image_processor", "depth": "depth_anything_image_processor",
"bae": "normalbae_image_processor", "bae": "normalbae_image_processor",
"normal_bae": "normalbae_image_processor", "normal": "normalbae_image_processor",
"sketch": "pidi_image_processor", "sketch": "pidi_image_processor",
"scribble": "lineart_image_processor", "scribble": "lineart_image_processor",
"lineart": "lineart_image_processor", "lineart": "lineart_image_processor",
"lineart_anime": "lineart_anime_image_processor", "lineart_anime": "lineart_anime_image_processor",
"softedge": "hed_image_processor", "softedge": "hed_image_processor",
"shuffle": "content_shuffle_image_processor", "shuffle": "content_shuffle_image_processor",
"openpose": "dw_openpose_image_processor", "pose": "dw_openpose_image_processor",
"mediapipe": "mediapipe_face_processor", "mediapipe": "mediapipe_face_processor",
"pidi": "pidi_image_processor", "pidi": "pidi_image_processor",
"zoe": "zoe_depth_image_processor", "zoe": "zoe_depth_image_processor",
"color": "color_map_image_processor", "color": "color_map_image_processor",
} }
def get_default_settings(self, model_name: str) -> Optional[ControlAdapterDefaultSettings]: @classmethod
if model_name in self.MODEL_NAME_TO_PREPROCESSOR: def get_default_settings(cls, model_name: str) -> Optional[ControlAdapterDefaultSettings]:
return ControlAdapterDefaultSettings(preprocessor=self.MODEL_NAME_TO_PREPROCESSOR[model_name]) for k, v in cls.MODEL_NAME_TO_PREPROCESSOR.items():
if k in model_name:
return ControlAdapterDefaultSettings(preprocessor=v)
return None return None