Spaces:
Runtime error
Runtime error
test / extensions-builtin /sd-webui-controlnet /scripts /preprocessor /legacy /preprocessor_compiled.py
| from .processor import * # noqa | |
| import functools | |
| legacy_preprocessors = { | |
| # "none": { | |
| # "label": "none", | |
| # "call_function": lambda x, *args, **kwargs: (x, True), | |
| # "unload_function": None, | |
| # "managed_model": None, | |
| # "model_free": False, | |
| # "no_control_mode": False, | |
| # "resolution": None, | |
| # "slider_1": None, | |
| # "slider_2": None, | |
| # "slider_3": None, | |
| # "priority": 100, | |
| # "tags": [] | |
| # }, | |
| # "invert": { | |
| # "label": "invert (from white bg & black line)", | |
| # "call_function": invert, | |
| # "unload_function": None, | |
| # "managed_model": None, | |
| # "model_free": False, | |
| # "no_control_mode": False, | |
| # "resolution": None, | |
| # "slider_1": None, | |
| # "slider_2": None, | |
| # "slider_3": None, | |
| # "priority": 20, | |
| # "tags": [ | |
| # "Canny", "Lineart", "Scribble", "MLSD", | |
| # ] | |
| # }, | |
| "animal_openpose": { | |
| "label": "animal_openpose", | |
| "call_function": functools.partial(g_openpose_model.run_model, include_body=True, include_hand=False, include_face=False, use_animal_pose=True), | |
| "unload_function": g_openpose_model.unload, | |
| "managed_model": "g_openpose_model", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": { | |
| "label": "Resolution", | |
| "minimum": 64, | |
| "maximum": 2048, | |
| "value": 512 | |
| }, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "OpenPose" | |
| ] | |
| }, | |
| # "blur_gaussian": { | |
| # "label": "blur_gaussian", | |
| # "call_function": blur_gaussian, | |
| # "unload_function": None, | |
| # "managed_model": None, | |
| # "model_free": False, | |
| # "no_control_mode": False, | |
| # "resolution": { | |
| # "label": "Resolution", | |
| # "value": 512, | |
| # "minimum": 64, | |
| # "maximum": 2048 | |
| # }, | |
| # "slider_1": { | |
| # "label": "Sigma", | |
| # "minimum": 0.01, | |
| # "maximum": 64.0, | |
| # "value": 9.0 | |
| # }, | |
| # "slider_2": None, | |
| # "slider_3": None, | |
| # "priority": 0, | |
| # "tags": [ | |
| # "Tile", | |
| # ] | |
| # }, | |
| # "canny": { | |
| # "label": "canny", | |
| # "call_function": canny, | |
| # "unload_function": None, | |
| # "managed_model": "model_canny", | |
| # "model_free": False, | |
| # "no_control_mode": False, | |
| # "resolution": { | |
| # "label": "Resolution", | |
| # "value": 512, | |
| # "minimum": 64, | |
| # "maximum": 2048 | |
| # }, | |
| # "slider_1": { | |
| # "label": "Canny Low Threshold", | |
| # "value": 100, | |
| # "minimum": 1, | |
| # "maximum": 255 | |
| # }, | |
| # "slider_2": { | |
| # "label": "Canny High Threshold", | |
| # "value": 200, | |
| # "minimum": 1, | |
| # "maximum": 255 | |
| # }, | |
| # "slider_3": None, | |
| # "priority": 100, | |
| # "tags": [ | |
| # "Canny" | |
| # ] | |
| # }, | |
| "densepose": { | |
| "label": "densepose (pruple bg & purple torso)", | |
| "call_function": functools.partial(densepose, cmap="viridis"), | |
| "unload_function": unload_densepose, | |
| "managed_model": "unknown", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": { | |
| "label": "Resolution", | |
| "minimum": 64, | |
| "maximum": 2048, | |
| "value": 512 | |
| }, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "OpenPose" | |
| ] | |
| }, | |
| "densepose_parula": { | |
| "label": "densepose_parula (black bg & blue torso)", | |
| "call_function": functools.partial(densepose, cmap="parula"), | |
| "unload_function": unload_densepose, | |
| "managed_model": "unknown", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": { | |
| "label": "Resolution", | |
| "minimum": 64, | |
| "maximum": 2048, | |
| "value": 512 | |
| }, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "OpenPose" | |
| ] | |
| }, | |
| "depth_anything": { | |
| "label": "depth_anything", | |
| "call_function": functools.partial(depth_anything, colored=False), | |
| "unload_function": unload_depth_anything, | |
| "managed_model": "model_depth_anything", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "Depth" | |
| ] | |
| }, | |
| "depth_hand_refiner": { | |
| "label": "depth_hand_refiner", | |
| "call_function": g_hand_refiner_model.run_model, | |
| "unload_function": g_hand_refiner_model.unload, | |
| "managed_model": "g_hand_refiner_model", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": { | |
| "label": "Resolution", | |
| "value": 512, | |
| "minimum": 64, | |
| "maximum": 2048 | |
| }, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "Depth" | |
| ] | |
| }, | |
| "depth_leres": { | |
| "label": "depth_leres", | |
| "call_function": functools.partial(leres, boost=False), | |
| "unload_function": unload_leres, | |
| "managed_model": "model_leres", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": { | |
| "label": "Resolution", | |
| "minimum": 64, | |
| "maximum": 2048, | |
| "value": 512 | |
| }, | |
| "slider_1": { | |
| "label": "Remove Near %", | |
| "minimum": 0, | |
| "maximum": 100, | |
| "value": 0, | |
| "step": 0.1 | |
| }, | |
| "slider_2": { | |
| "label": "Remove Background %", | |
| "minimum": 0, | |
| "maximum": 100, | |
| "value": 0, | |
| "step": 0.1 | |
| }, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "Depth" | |
| ] | |
| }, | |
| "depth_leres++": { | |
| "label": "depth_leres++", | |
| "call_function": functools.partial(leres, boost=True), | |
| "unload_function": unload_leres, | |
| "managed_model": "model_leres", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": { | |
| "label": "Resolution", | |
| "minimum": 64, | |
| "maximum": 2048, | |
| "value": 512 | |
| }, | |
| "slider_1": { | |
| "label": "Remove Near %", | |
| "minimum": 0, | |
| "maximum": 100, | |
| "value": 0, | |
| "step": 0.1 | |
| }, | |
| "slider_2": { | |
| "label": "Remove Background %", | |
| "minimum": 0, | |
| "maximum": 100, | |
| "value": 0, | |
| "step": 0.1 | |
| }, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "Depth" | |
| ] | |
| }, | |
| "depth": { | |
| "label": "depth_midas", | |
| "call_function": midas, | |
| "unload_function": unload_midas, | |
| "managed_model": "model_midas", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": { | |
| "label": "Resolution", | |
| "minimum": 64, | |
| "maximum": 2048, | |
| "value": 512 | |
| }, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 100, | |
| "tags": [ | |
| "Depth" | |
| ] | |
| }, | |
| "depth_zoe": { | |
| "label": "depth_zoe", | |
| "call_function": zoe_depth, | |
| "unload_function": unload_zoe_depth, | |
| "managed_model": "model_zoe_depth", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "Depth" | |
| ] | |
| }, | |
| "dw_openpose_full": { | |
| "label": "dw_openpose_full", | |
| "call_function": functools.partial(g_openpose_model.run_model, include_body=True, include_hand=True, include_face=True, use_dw_pose=True), | |
| "unload_function": g_openpose_model.unload, | |
| "managed_model": 'g_openpose_model', | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": { | |
| "label": "Resolution", | |
| "minimum": 64, | |
| "maximum": 2048, | |
| "value": 512 | |
| }, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "OpenPose" | |
| ] | |
| }, | |
| "inpaint": { | |
| "label": "inpaint_global_harmonious", | |
| "call_function": identity, | |
| "unload_function": None, | |
| "managed_model": None, | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "Inpaint" | |
| ] | |
| }, | |
| "inpaint_only": { | |
| "label": "inpaint_only", | |
| "call_function": identity, | |
| "unload_function": None, | |
| "managed_model": None, | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 100, | |
| "tags": [ | |
| "Inpaint" | |
| ] | |
| }, | |
| # "inpaint_only+lama": { | |
| # "label": "inpaint_only+lama", | |
| # "call_function": lama_inpaint, | |
| # "unload_function": unload_lama_inpaint, | |
| # "managed_model": "model_lama", | |
| # "model_free": False, | |
| # "no_control_mode": False, | |
| # "resolution": None, | |
| # "slider_1": None, | |
| # "slider_2": None, | |
| # "slider_3": None, | |
| # "priority": 0, | |
| # "tags": [ | |
| # "Inpaint" | |
| # ] | |
| # }, | |
| "instant_id_face_embedding": { | |
| "label": "instant_id_face_embedding", | |
| "call_function": functools.partial(g_insight_face_instant_id_model.run_model_instant_id, return_keypoints=False), | |
| "unload_function": None, | |
| "managed_model": None, | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "Instant-ID" | |
| ], | |
| "returns_image": False, | |
| }, | |
| "instant_id_face_keypoints": { | |
| "label": "instant_id_face_keypoints", | |
| "call_function": functools.partial(g_insight_face_instant_id_model.run_model_instant_id, return_keypoints=True), | |
| "unload_function": None, | |
| "managed_model": None, | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "Instant-ID" | |
| ] | |
| }, | |
| "ip-adapter_clip_sd15": { | |
| "label": "ip-adapter_clip_h", | |
| "call_function": functools.partial(clip, config='clip_h'), | |
| "unload_function": functools.partial(unload_clip, config='clip_h'), | |
| "managed_model": "unknown", | |
| "model_free": False, | |
| "no_control_mode": True, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 100, | |
| "tags": [ | |
| "IP-Adapter" | |
| ], | |
| "returns_image": False, | |
| }, | |
| "ip-adapter_clip_sdxl": { | |
| "label": "ip-adapter_clip_g", | |
| "call_function": functools.partial(clip, config='clip_g'), | |
| "unload_function": functools.partial(unload_clip, config='clip_g'), | |
| "managed_model": "unknown", | |
| "model_free": False, | |
| "no_control_mode": True, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "IP-Adapter" | |
| ], | |
| "returns_image": False, | |
| }, | |
| "ip-adapter_clip_sdxl_plus_vith": { | |
| "label": "ip-adapter_clip_sdxl_plus_vith", | |
| "call_function": functools.partial(clip, config='clip_h'), | |
| "unload_function": functools.partial(unload_clip, config='clip_h'), | |
| "managed_model": "unknown", | |
| "model_free": False, | |
| "no_control_mode": True, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "IP-Adapter" | |
| ], | |
| "returns_image": False, | |
| }, | |
| "ip-adapter_face_id": { | |
| "label": "ip-adapter_face_id", | |
| "call_function": g_insight_face_model.run_model, | |
| "unload_function": None, | |
| "managed_model": None, | |
| "model_free": False, | |
| "no_control_mode": True, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "IP-Adapter" | |
| ], | |
| "returns_image": False, | |
| }, | |
| "ip-adapter_face_id_plus": { | |
| "label": "ip-adapter_face_id_plus", | |
| "call_function": face_id_plus, | |
| "unload_function": functools.partial(unload_clip, config='clip_h'), | |
| "managed_model": "unknown", | |
| "model_free": False, | |
| "no_control_mode": True, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "IP-Adapter" | |
| ], | |
| "returns_image": False, | |
| }, | |
| "lineart_anime": { | |
| "label": "lineart_anime", | |
| "call_function": lineart_anime, | |
| "unload_function": unload_lineart_anime, | |
| "managed_model": "model_lineart_anime", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "Lineart" | |
| ] | |
| }, | |
| "lineart_anime_denoise": { | |
| "label": "lineart_anime_denoise", | |
| "call_function": lineart_anime_denoise, | |
| "unload_function": unload_lineart_anime_denoise, | |
| "managed_model": "model_manga_line", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "Lineart" | |
| ] | |
| }, | |
| "lineart_coarse": { | |
| "label": "lineart_coarse", | |
| "call_function": lineart_coarse, | |
| "unload_function": unload_lineart_coarse, | |
| "managed_model": "model_lineart_coarse", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "Lineart" | |
| ] | |
| }, | |
| "lineart": { | |
| "label": "lineart_realistic", | |
| "call_function": lineart, | |
| "unload_function": unload_lineart, | |
| "managed_model": "model_lineart", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "Lineart" | |
| ] | |
| }, | |
| "lineart_standard": { | |
| "label": "lineart_standard (from white bg & black line)", | |
| "call_function": lineart_standard, | |
| "unload_function": None, | |
| "managed_model": None, | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 100, | |
| "tags": [ | |
| "Lineart" | |
| ] | |
| }, | |
| "mediapipe_face": { | |
| "label": "mediapipe_face", | |
| "call_function": mediapipe_face, | |
| "unload_function": None, | |
| "managed_model": None, | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": { | |
| "label": "Resolution", | |
| "value": 512, | |
| "minimum": 64, | |
| "maximum": 2048 | |
| }, | |
| "slider_1": { | |
| "label": "Max Faces", | |
| "value": 1, | |
| "minimum": 1, | |
| "maximum": 10, | |
| "step": 1 | |
| }, | |
| "slider_2": { | |
| "label": "Min Face Confidence", | |
| "value": 0.5, | |
| "minimum": 0.01, | |
| "maximum": 1.0, | |
| "step": 0.01 | |
| }, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [] | |
| }, | |
| "mlsd": { | |
| "label": "mlsd", | |
| "call_function": mlsd, | |
| "unload_function": unload_mlsd, | |
| "managed_model": "model_mlsd", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": { | |
| "label": "Resolution", | |
| "minimum": 64, | |
| "maximum": 2048, | |
| "value": 512 | |
| }, | |
| "slider_1": { | |
| "label": "MLSD Value Threshold", | |
| "minimum": 0.01, | |
| "maximum": 2.0, | |
| "value": 0.1, | |
| "step": 0.01 | |
| }, | |
| "slider_2": { | |
| "label": "MLSD Distance Threshold", | |
| "minimum": 0.01, | |
| "maximum": 20.0, | |
| "value": 0.1, | |
| "step": 0.01 | |
| }, | |
| "slider_3": None, | |
| "priority": 100, | |
| "tags": [ | |
| "MLSD" | |
| ], | |
| "use_soft_projection_in_hr_fix": True | |
| }, | |
| "normal_bae": { | |
| "label": "normal_bae", | |
| "call_function": normal_bae, | |
| "unload_function": unload_normal_bae, | |
| "managed_model": "model_normal_bae", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 100, | |
| "tags": [ | |
| "NormalMap" | |
| ] | |
| }, | |
| "normal_map": { | |
| "label": "normal_midas", | |
| "call_function": midas_normal, | |
| "unload_function": unload_midas, | |
| "managed_model": "model_midas", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": { | |
| "label": "Resolution", | |
| "minimum": 64, | |
| "maximum": 2048, | |
| "value": 512 | |
| }, | |
| "slider_1": { | |
| "label": "Normal Background Threshold", | |
| "minimum": 0.0, | |
| "maximum": 1.0, | |
| "value": 0.4, | |
| "step": 0.01 | |
| }, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "NormalMap" | |
| ] | |
| }, | |
| "openpose": { | |
| "label": "openpose", | |
| "call_function": functools.partial(g_openpose_model.run_model, include_body=True, include_hand=False, include_face=False), | |
| "unload_function": g_openpose_model.unload, | |
| "managed_model": "g_openpose_model", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": { | |
| "label": "Resolution", | |
| "minimum": 64, | |
| "maximum": 2048, | |
| "value": 512 | |
| }, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "OpenPose" | |
| ] | |
| }, | |
| "openpose_face": { | |
| "label": "openpose_face", | |
| "call_function": functools.partial(g_openpose_model.run_model, include_body=True, include_hand=False, include_face=True), | |
| "unload_function": g_openpose_model.unload, | |
| "managed_model": "g_openpose_model", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "OpenPose" | |
| ] | |
| }, | |
| "openpose_faceonly": { | |
| "label": "openpose_faceonly", | |
| "call_function": functools.partial(g_openpose_model.run_model, include_body=False, include_hand=False, include_face=True), | |
| "unload_function": g_openpose_model.unload, | |
| "managed_model": "g_openpose_model", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "OpenPose" | |
| ] | |
| }, | |
| "openpose_full": { | |
| "label": "openpose_full", | |
| "call_function": functools.partial(g_openpose_model.run_model, include_body=True, include_hand=True, include_face=True), | |
| "unload_function": g_openpose_model.unload, | |
| "managed_model": "g_openpose_model", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": { | |
| "label": "Resolution", | |
| "minimum": 64, | |
| "maximum": 2048, | |
| "value": 512 | |
| }, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 100, | |
| "tags": [ | |
| "OpenPose" | |
| ] | |
| }, | |
| "openpose_hand": { | |
| "label": "openpose_hand", | |
| "call_function": functools.partial(g_openpose_model.run_model, include_body=True, include_hand=True, include_face=False), | |
| "unload_function": g_openpose_model.unload, | |
| "managed_model": "g_openpose_model", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "OpenPose" | |
| ] | |
| }, | |
| "recolor_intensity": { | |
| "label": "recolor_intensity", | |
| "call_function": recolor_intensity, | |
| "unload_function": None, | |
| "managed_model": None, | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": { | |
| "label": "Gamma Correction", | |
| "value": 1.0, | |
| "minimum": 0.1, | |
| "maximum": 2.0, | |
| "step": 0.001 | |
| }, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "Recolor" | |
| ] | |
| }, | |
| "recolor_luminance": { | |
| "label": "recolor_luminance", | |
| "call_function": recolor_luminance, | |
| "unload_function": None, | |
| "managed_model": None, | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": { | |
| "label": "Gamma Correction", | |
| "value": 1.0, | |
| "minimum": 0.1, | |
| "maximum": 2.0, | |
| "step": 0.001 | |
| }, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 100, | |
| "tags": [ | |
| "Recolor" | |
| ] | |
| }, | |
| "reference_adain": { | |
| "label": "reference_adain", | |
| "call_function": identity, | |
| "unload_function": None, | |
| "managed_model": None, | |
| "model_free": True, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": { | |
| "label": "Style Fidelity (only for Balanced mode)", | |
| "value": 0.5, | |
| "minimum": 0.0, | |
| "maximum": 1.0, | |
| "step": 0.01 | |
| }, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "Reference" | |
| ] | |
| }, | |
| "reference_adain+attn": { | |
| "label": "reference_adain+attn", | |
| "call_function": identity, | |
| "unload_function": None, | |
| "managed_model": None, | |
| "model_free": True, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": { | |
| "label": "Style Fidelity (only for Balanced mode)", | |
| "value": 0.5, | |
| "minimum": 0.0, | |
| "maximum": 1.0, | |
| "step": 0.01 | |
| }, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "Reference" | |
| ] | |
| }, | |
| "reference_only": { | |
| "label": "reference_only", | |
| "call_function": identity, | |
| "unload_function": None, | |
| "managed_model": None, | |
| "model_free": True, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": { | |
| "label": "Style Fidelity (only for Balanced mode)", | |
| "value": 0.5, | |
| "minimum": 0.0, | |
| "maximum": 1.0, | |
| "step": 0.01 | |
| }, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 100, | |
| "tags": [ | |
| "Reference" | |
| ] | |
| }, | |
| "revision_clipvision": { | |
| "label": "revision_clipvision", | |
| "call_function": functools.partial(clip, config='clip_g'), | |
| "unload_function": functools.partial(unload_clip, config='clip_g'), | |
| "managed_model": None, | |
| "model_free": True, | |
| "no_control_mode": True, | |
| "resolution": None, | |
| "slider_1": { | |
| "label": "Noise Augmentation", | |
| "value": 0.0, | |
| "minimum": 0.0, | |
| "maximum": 1.0 | |
| }, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 100, | |
| "tags": [ | |
| "Revision" | |
| ], | |
| "returns_image": False, | |
| }, | |
| "revision_ignore_prompt": { | |
| "label": "revision_ignore_prompt", | |
| "call_function": functools.partial(clip, config='clip_g'), | |
| "unload_function": functools.partial(unload_clip, config='clip_g'), | |
| "managed_model": None, | |
| "model_free": True, | |
| "no_control_mode": True, | |
| "resolution": None, | |
| "slider_1": { | |
| "label": "Noise Augmentation", | |
| "value": 0.0, | |
| "minimum": 0.0, | |
| "maximum": 1.0 | |
| }, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "Revision" | |
| ], | |
| "returns_image": False, | |
| }, | |
| "scribble_hed": { | |
| "label": "scribble_hed", | |
| "call_function": scribble_hed, | |
| "unload_function": unload_hed, | |
| "managed_model": "model_hed", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": { | |
| "label": "Resolution", | |
| "minimum": 64, | |
| "maximum": 2048, | |
| "value": 512 | |
| }, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "Scribble", "SparseCtrl", | |
| ] | |
| }, | |
| "pidinet_scribble": { | |
| "label": "scribble_pidinet", | |
| "call_function": scribble_pidinet, | |
| "unload_function": None, | |
| "managed_model": None, | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 100, | |
| "tags": [ | |
| "Scribble", "SparseCtrl", | |
| ] | |
| }, | |
| # "scribble_xdog": { | |
| # "label": "scribble_xdog", | |
| # "call_function": scribble_xdog, | |
| # "unload_function": None, | |
| # "managed_model": None, | |
| # "model_free": False, | |
| # "no_control_mode": False, | |
| # "resolution": { | |
| # "label": "Resolution", | |
| # "value": 512, | |
| # "minimum": 64, | |
| # "maximum": 2048 | |
| # }, | |
| # "slider_1": { | |
| # "label": "XDoG Threshold", | |
| # "minimum": 1, | |
| # "maximum": 64, | |
| # "value": 32 | |
| # }, | |
| # "slider_2": None, | |
| # "slider_3": None, | |
| # "priority": 0, | |
| # "tags": [ | |
| # "Scribble", | |
| # ] | |
| # }, | |
| "anime_face_segment": { | |
| "label": "seg_anime_face", | |
| "call_function": anime_face_segment, | |
| "unload_function": unload_anime_face_segment, | |
| "managed_model": "model_anime_face_segment", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": { | |
| "label": "Resolution", | |
| "value": 512, | |
| "minimum": 64, | |
| "maximum": 2048 | |
| }, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "Segmentation" | |
| ] | |
| }, | |
| "oneformer_ade20k": { | |
| "label": "seg_ofade20k", | |
| "call_function": oneformer_ade20k, | |
| "unload_function": unload_oneformer_ade20k, | |
| "managed_model": "model_oneformer_ade20k", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 100, | |
| "tags": [ | |
| "Segmentation" | |
| ] | |
| }, | |
| "oneformer_coco": { | |
| "label": "seg_ofcoco", | |
| "call_function": oneformer_coco, | |
| "unload_function": unload_oneformer_coco, | |
| "managed_model": "model_oneformer_coco", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "Segmentation" | |
| ] | |
| }, | |
| "segmentation": { | |
| "label": "seg_ufade20k", | |
| "call_function": uniformer, | |
| "unload_function": unload_uniformer, | |
| "managed_model": "model_uniformer", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": { | |
| "label": "Resolution", | |
| "minimum": 64, | |
| "maximum": 2048, | |
| "value": 512 | |
| }, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "Segmentation" | |
| ] | |
| }, | |
| # "shuffle": { | |
| # "label": "shuffle", | |
| # "call_function": shuffle, | |
| # "unload_function": None, | |
| # "managed_model": None, | |
| # "model_free": False, | |
| # "no_control_mode": False, | |
| # "resolution": None, | |
| # "slider_1": None, | |
| # "slider_2": None, | |
| # "slider_3": None, | |
| # "priority": 100, | |
| # "tags": [ | |
| # "Shuffle" | |
| # ] | |
| # }, | |
| "hed": { | |
| "label": "softedge_hed", | |
| "call_function": hed, | |
| "unload_function": unload_hed, | |
| "managed_model": "model_hed", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": { | |
| "label": "Resolution", | |
| "minimum": 64, | |
| "maximum": 2048, | |
| "value": 512 | |
| }, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "SoftEdge" | |
| ] | |
| }, | |
| "hed_safe": { | |
| "label": "softedge_hedsafe", | |
| "call_function": hed_safe, | |
| "unload_function": unload_hed, | |
| "managed_model": "model_hed", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": { | |
| "label": "Resolution", | |
| "minimum": 64, | |
| "maximum": 2048, | |
| "value": 512 | |
| }, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "SoftEdge" | |
| ] | |
| }, | |
| "pidinet": { | |
| "label": "softedge_pidinet", | |
| "call_function": pidinet, | |
| "unload_function": unload_pidinet, | |
| "managed_model": "model_pidinet", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 100, | |
| "tags": [ | |
| "SoftEdge" | |
| ] | |
| }, | |
| "pidinet_safe": { | |
| "label": "softedge_pidisafe", | |
| "call_function": pidinet_safe, | |
| "unload_function": unload_pidinet, | |
| "managed_model": "model_pidinet", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "SoftEdge" | |
| ] | |
| }, | |
| "te_hed": { | |
| "label": "softedge_teed", | |
| "call_function": te_hed, | |
| "unload_function": unload_te_hed, | |
| "managed_model": "model_te_hed", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": { | |
| "label": "Resolution", | |
| "value": 512, | |
| "minimum": 64, | |
| "maximum": 2048 | |
| }, | |
| "slider_1": { | |
| "label": "Safe Steps", | |
| "minimum": 0, | |
| "maximum": 10, | |
| "value": 2, | |
| "step": 1 | |
| }, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "SoftEdge" | |
| ] | |
| }, | |
| "color": { | |
| "label": "t2ia_color_grid", | |
| "call_function": color, | |
| "unload_function": None, | |
| "managed_model": None, | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": { | |
| "label": "Resolution", | |
| "value": 512, | |
| "minimum": 64, | |
| "maximum": 2048 | |
| }, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "T2I-Adapter" | |
| ] | |
| }, | |
| "pidinet_sketch": { | |
| "label": "t2ia_sketch_pidi", | |
| "call_function": pidinet_ts, | |
| "unload_function": unload_pidinet, | |
| "managed_model": "model_pidinet", | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "T2I-Adapter" | |
| ] | |
| }, | |
| "clip_vision": { | |
| "label": "t2ia_style_clipvision", | |
| "call_function": functools.partial(clip, config='clip_vitl'), | |
| "unload_function": functools.partial(unload_clip, config='clip_vitl'), | |
| "managed_model": "unknown", | |
| "model_free": False, | |
| "no_control_mode": True, | |
| "resolution": None, | |
| "slider_1": None, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "T2I-Adapter" | |
| ], | |
| "returns_image": False, | |
| }, | |
| "threshold": { | |
| "label": "threshold", | |
| "call_function": threshold, | |
| "unload_function": None, | |
| "managed_model": None, | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": { | |
| "label": "Resolution", | |
| "value": 512, | |
| "minimum": 64, | |
| "maximum": 2048 | |
| }, | |
| "slider_1": { | |
| "label": "Binarization Threshold", | |
| "minimum": 0, | |
| "maximum": 255, | |
| "value": 127 | |
| }, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [] | |
| }, | |
| "tile_colorfix": { | |
| "label": "tile_colorfix", | |
| "call_function": identity, | |
| "unload_function": None, | |
| "managed_model": None, | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": { | |
| "label": "Variation", | |
| "value": 8.0, | |
| "minimum": 3.0, | |
| "maximum": 32.0, | |
| "step": 1.0 | |
| }, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "Tile", | |
| ] | |
| }, | |
| "tile_colorfix+sharp": { | |
| "label": "tile_colorfix+sharp", | |
| "call_function": identity, | |
| "unload_function": None, | |
| "managed_model": None, | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": { | |
| "label": "Variation", | |
| "value": 8.0, | |
| "minimum": 3.0, | |
| "maximum": 32.0, | |
| "step": 1.0 | |
| }, | |
| "slider_2": { | |
| "label": "Sharpness", | |
| "value": 1.0, | |
| "minimum": 0.0, | |
| "maximum": 2.0, | |
| "step": 0.01 | |
| }, | |
| "slider_3": None, | |
| "priority": 0, | |
| "tags": [ | |
| "Tile", | |
| ] | |
| }, | |
| "tile_resample": { | |
| "label": "tile_resample", | |
| "call_function": tile_resample, | |
| "unload_function": None, | |
| "managed_model": None, | |
| "model_free": False, | |
| "no_control_mode": False, | |
| "resolution": None, | |
| "slider_1": { | |
| "label": "Down Sampling Rate", | |
| "value": 1.0, | |
| "minimum": 1.0, | |
| "maximum": 8.0, | |
| "step": 0.01 | |
| }, | |
| "slider_2": None, | |
| "slider_3": None, | |
| "priority": 100, | |
| "tags": [ | |
| "Tile", | |
| ] | |
| } | |
| } | |