-
Notifications
You must be signed in to change notification settings - Fork 7
/
__init__.py
73 lines (66 loc) · 2.44 KB
/
__init__.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
"""
@author: Pos13
@title: Cyclist
@nickname: comfyui-cyclist
@description: This extension provides tools to iterate generation results between runs. In general, it's for cycles.
"""
#import inspect
from .cyclist import *
from .util_nodes import *
WEB_DIRECTORY = "./js"
NODE_CLASS_MAPPINGS = {
"LoopManager": LoopManager,
"Interrupt": CyclistInterrupt,
"ReloadImage": ReloadImage,
"OverrideImage": OverrideImage,
"ReloadLatent": ReloadLatent,
"OverrideLatent": OverrideLatent,
"ReloadModel": ReloadModel,
"OverrideModel": OverrideModel,
"RecallString": RecallString,
"MemorizeString": MemorizeString,
"RecallInt": RecallInt,
"MemorizeInt": MemorizeInt,
"RecallFloat": RecallFloat,
"MemorizeFloat": MemorizeFloat,
"RecallConditioning": RecallConditioning,
"MemorizeConditioning": MemorizeConditioning,
"CyclistMathInt": CyclistMathInt,
"CyclistMathFloat": CyclistMathFloat,
"CyclistTypeCast": CyclistTypeCast,
"CyclistCompare": CyclistCompare,
"CyclistTimer": CyclistTimer,
"CyclistTimerStop": CyclistTimerStop
}
NODE_DISPLAY_NAME_MAPPINGS = {
"LoopManager": "Loop Manager",
"Interrupt": "Interrupt",
"RecallString": "Recall String",
"MemorizeString": "Memorize String",
"RecallInt": "Recall Int",
"MemorizeInt": "Memorize Int",
"RecallFloat": "Recall Float",
"MemorizeFloat": "Memorize Float",
"RecallConditioning": "Recall Conditioning",
"MemorizeConditioning": "Memorize Conditioning",
"ReloadLatent": "Reload Latent",
"OverrideLatent": "Save Latent (Override)",
"ReloadImage": "Reload Image",
"OverrideImage": "Save Image (Override)",
"ReloadModel": "Reload Model",
"OverrideModel": "Save Model (Override)",
"CyclistTimer": "Generation Timer",
"CyclistTimerStop": "Force Timer Stop",
"CyclistMathFloat": "Float Math",
"CyclistMathInt": "Int Math",
"CyclistTypeCast": "Convert to",
"CyclistCompare": "Compare Anything"
}
# Sadly, this automatic mapping is abandoned for Manager compatibility
"""NODE_CLASS_MAPPINGS = {}
for name, obj in (list(reversed(inspect.getmembers(cyclist))) + inspect.getmembers(util_nodes)):
if inspect.isclass(obj):
if hasattr(obj, 'NODE_NAME'):
node_name = getattr(obj, 'NODE_NAME', name)
NODE_CLASS_MAPPINGS[node_name] = obj"""
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', 'WEB_DIRECTORY']