-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy path_nodes.py
85 lines (83 loc) · 4.75 KB
/
_nodes.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
from .nodes.crop_resize import RL_Crop_Resize, RL_Uncrop, RL_BBox
from .nodes.image_shadow import RL_Image_Shadow
from .nodes.optical_flow import RL_Optical_Flow_Dip, RL_Warp_Image, RL_Save_Flow, RL_Load_Flow
from .nodes.image_threshold import RL_Image_Threshold_Channels
from .nodes.finetune import RL_Finetune_Variable, RL_Finetune_Analyze, RL_Finetune_Analyze_Batch
from .nodes.files import RL_SaveImageSequence, RL_LoadImageSequence, RL_IfFileExists, RL_CacheImageSequence, RL_CacheMaskSequence, RL_SequenceContext, RL_Sequence_ToFilePathList
from .nodes.depth_16 import RL_Zoe_Depth_Map_Preprocessor, RL_Zoe_Depth_Map_Preprocessor_Raw_Infer, RL_Zoe_Depth_Map_Preprocessor_Raw_Process
from .nodes.internet_search import RL_Internet_Search
from .nodes.dependency_order import RL_ForceDependencyOrder, RL_ForceDependencyOrder_Latents, RL_ForceDependencyOrder_Images, RL_ForceDependencyOrder_String, RL_ForceDependencyOrder_Strings, RL_ForceDependencyOrder_ImageString, RL_StopIfBlack, RL_RebootComfyIfLeaky
from .nodes.lora_tools import RL_LoraTextExtractTags
from .nodes.civitai_tools import RL_CivitaiTopImagePrompts
NODE_CLASS_MAPPINGS = {
"RL_BBox": RL_BBox,
"RL_Crop_Resize": RL_Crop_Resize,
"RL_Uncrop": RL_Uncrop,
# "RL_Crop_Resize_Batch": RL_Crop_Resize_Batch,
"RL_Image_Shadow": RL_Image_Shadow,
"RL_Optical_Flow_Dip": RL_Optical_Flow_Dip,
"RL_Warp_Image": RL_Warp_Image,
"RL_Save_Flow": RL_Save_Flow,
"RL_Load_Flow": RL_Load_Flow,
"RL_Image_Threshold_Channels": RL_Image_Threshold_Channels,
"RL_Finetune_Variable": RL_Finetune_Variable,
"RL_Finetune_Analyze": RL_Finetune_Analyze,
"RL_Finetune_Analyze_Batch": RL_Finetune_Analyze_Batch,
"RL_SaveImageSequence": RL_SaveImageSequence,
"RL_LoadImageSequence": RL_LoadImageSequence,
"RL_IfFileExists": RL_IfFileExists,
"RL_CacheImageSequence": RL_CacheImageSequence,
"RL_CacheMaskSequence": RL_CacheMaskSequence,
"RL_SequenceContext": RL_SequenceContext,
"RL_Sequence_ToFilePathList": RL_Sequence_ToFilePathList,
"RL_Zoe_Depth_Map_Preprocessor": RL_Zoe_Depth_Map_Preprocessor,
"RL_Zoe_Depth_Map_Preprocessor_Raw_Infer": RL_Zoe_Depth_Map_Preprocessor_Raw_Infer,
"RL_Zoe_Depth_Map_Preprocessor_Raw_Process": RL_Zoe_Depth_Map_Preprocessor_Raw_Process,
"RL_Internet_Search": RL_Internet_Search,
"RL_ForceDependencyOrder": RL_ForceDependencyOrder,
"RL_ForceDependencyOrder_Latents": RL_ForceDependencyOrder_Latents,
"RL_ForceDependencyOrder_Images": RL_ForceDependencyOrder_Images,
"RL_ForceDependencyOrder_String": RL_ForceDependencyOrder_String,
"RL_ForceDependencyOrder_Strings": RL_ForceDependencyOrder_Strings,
"RL_ForceDependencyOrder_ImageString": RL_ForceDependencyOrder_ImageString,
"RL_RebootComfyIfLeaky": RL_RebootComfyIfLeaky,
"RL_StopIfBlack": RL_StopIfBlack,
"RL_LoraTextExtractTags": RL_LoraTextExtractTags,
"RL_CivitaiTopImagePrompts": RL_CivitaiTopImagePrompts,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"RL_BBox": "RL BBox",
"RL_Crop_Resize": "RL Image Crop and Resize by Mask",
"RL_Uncrop": "RL Image Uncrop",
# "RL_Crop_Resize_Batch": "Image Crop and Resize by Mask (Batch)",
"RL_Image_Shadow": "RL Image Shadow",
"RL_Optical_Flow_Dip": "RL Optical Flow (Dip)",
"RL_Warp_Image": "RL Warp Image with Flow",
"RL_Save_Flow": "RL Save Flow",
"RL_Load_Flow": "RL Load Flow",
"RL_Image_Threshold_Channels": "RL Image Threshold (Channels)",
"RL_Finetune_Variable": "RL Finetune Variable",
"RL_Finetune_Analyze": "RL Finetune Analyze",
"RL_Finetune_Analyze_Batch": "RL Finetune Analyze Batch",
"RL_SaveImageSequence": "RL SaveImageSequence",
"RL_LoadImageSequence": "RL LoadImageSequence",
"RL_IfFileExists": "RL IfFileExists",
"RL_CacheImageSequence": "RL CacheImageSequence",
"RL_CacheMaskSequence": "RL CacheMaskSequence",
"RL_SequenceContext": "RL SequenceContext",
"RL_Sequence_ToFilePathList": "RL Sequence ToFilePathList",
"RL_Zoe_Depth_Map_Preprocessor": "RL Zoe Depth Map Preprocessor",
"RL_Zoe_Depth_Map_Preprocessor_Raw_Infer": "RL Zoe Depth Map Preprocessor Raw Infer",
"RL_Zoe_Depth_Map_Preprocessor_Raw_Process": "RL Zoe Depth Map Preprocessor Raw Process",
"RL_Internet_Search": "RL Internet Search",
"RL_ForceDependencyOrder": "RL Force",
"RL_ForceDependencyOrder_Latents": "RL Force Latents",
"RL_ForceDependencyOrder_Images": "RL Force Images",
"RL_ForceDependencyOrder_String": "RL Force String",
"RL_ForceDependencyOrder_Strings": "RL Force Strings",
"RL_ForceDependencyOrder_ImageString": "RL Force ImageString",
"RL_StopIfBlack": "RL StopIfBlack",
"RL_RebootComfyIfLeaky": "RL RebootComfyIfLeaky",
"RL_LoraTextExtractTags": "RL LoraTextExtractTags",
"RL_CivitaiTopImagePrompts": "RL CivitaiTopImagePrompts",
}