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

feat: Update ControlNet Model List & Map

Browse Source
This commit is contained in:
blessedcoolant 2023-06-03 18:21:06 +12:00 committed by psychedelicious
parent 73a95973a8
commit aa3a969bd2
1 changed files with 26 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
invokeai/frontend/web/src/features/controlNet/store/constants.ts
View File

@ -166,14 +166,21 @@ export const CONTROLNET_PROCESSORS: ControlNetProcessorsDict = {
}; };
export const CONTROLNET_MODELS = [ export const CONTROLNET_MODELS = [
'lllyasviel/sd-controlnet-canny', 'lllyasviel/control_v11p_sd15_canny',
'lllyasviel/sd-controlnet-depth', 'lllyasviel/control_v11p_sd15_inpaint',
'lllyasviel/sd-controlnet-hed', 'lllyasviel/control_v11p_sd15_mlsd',
'lllyasviel/sd-controlnet-seg', 'lllyasviel/control_v11f1p_sd15_depth',
'lllyasviel/sd-controlnet-openpose', 'lllyasviel/control_v11p_sd15_normalbae',
'lllyasviel/sd-controlnet-scribble', 'lllyasviel/control_v11p_sd15_seg',
'lllyasviel/sd-controlnet-normal', 'lllyasviel/control_v11p_sd15_lineart',
'lllyasviel/sd-controlnet-mlsd', 'lllyasviel/control_v11p_sd15s2_lineart_anime',
'lllyasviel/control_v11p_sd15_scribble',
'lllyasviel/control_v11p_sd15_softedge',
'lllyasviel/control_v11e_sd15_shuffle',
'lllyasviel/control_v11p_sd15_openpose',
'lllyasviel/control_v11f1e_sd15_tile',
'lllyasviel/control_v11e_sd15_ip2p',
'CrucibleAI/ControlNetMediaPipeFace',
]; ];
export type ControlNetModel = (typeof CONTROLNET_MODELS)[number]; export type ControlNetModel = (typeof CONTROLNET_MODELS)[number];
@ -182,9 +189,15 @@ export const CONTROLNET_MODEL_MAP: Record<
ControlNetModel, ControlNetModel,
ControlNetProcessorType ControlNetProcessorType
> = { > = {
'lllyasviel/sd-controlnet-canny': 'canny_image_processor', 'lllyasviel/control_v11p_sd15_canny': 'canny_image_processor',
'lllyasviel/sd-controlnet-depth': 'midas_depth_image_processor', 'lllyasviel/control_v11p_sd15_mlsd': 'mlsd_image_processor',
'lllyasviel/sd-controlnet-hed': 'hed_image_processor', 'lllyasviel/control_v11f1p_sd15_depth': 'midas_depth_image_processor',
'lllyasviel/sd-controlnet-openpose': 'openpose_image_processor', 'lllyasviel/control_v11p_sd15_normalbae': 'normalbae_image_processor',
'lllyasviel/sd-controlnet-mlsd': 'mlsd_image_processor', 'lllyasviel/control_v11p_sd15_lineart': 'lineart_image_processor',
'lllyasviel/control_v11p_sd15s2_lineart_anime':
'lineart_anime_image_processor',
'lllyasviel/control_v11p_sd15_softedge': 'hed_image_processor',
'lllyasviel/control_v11e_sd15_shuffle': 'content_shuffle_image_processor',
'lllyasviel/control_v11p_sd15_openpose': 'openpose_image_processor',
'CrucibleAI/ControlNetMediaPipeFace': 'mediapipe_face_processor',
}; };