-
-
Notifications
You must be signed in to change notification settings - Fork 5
Expand file tree
/
Copy path__init__.py
More file actions
32 lines (30 loc) · 1.19 KB
/
Copy path__init__.py
File metadata and controls
32 lines (30 loc) · 1.19 KB
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
from .try_off_nodes import ( # noqa
FluxFillPipelineNode,
TryOffFluxFillModelNode,
TryOffQuantizerNode,
TryOffRunNode,
TryOnOffModelNode,
TryOnOffRunNode,
TryOnRunNode,
)
# Node Registrations
NODE_CLASS_MAPPINGS = {
"TryOffModelNode": TryOnOffModelNode,
"TryOnOffModelNode": TryOnOffModelNode,
"TryOffFluxFillModelNode": TryOffFluxFillModelNode,
"TryOffRunNode": TryOffRunNode,
"TryOffQuantizerNode": TryOffQuantizerNode,
"TryOffFluxFillPipelineNode": FluxFillPipelineNode,
"TryOnRunNode": TryOnRunNode,
"TryOnOffRunNode": TryOnOffRunNode,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"TryOffModelNode": "TryOff Model Loader [ComfyUI-Flux-TryOff]",
"TryOnOffModelNode": "TryOnOff Model Loader [ComfyUI-Flux-TryOff]",
"TryOffFluxFillModelNode": "FluxFill Model Loader [ComfyUI-Flux-TryOff]",
"TryOffRunNode": "Run TryOff Inference [ComfyUI-Flux-TryOff]",
"TryOffQuantizerNode": "TryOff Quantizer [ComfyUI-Flux-TryOff]",
"TryOffFluxFillPipelineNode": "FluxFill Pipeline Loader [ComfyUI-Flux-TryOff]",
"TryOnRunNode": "Run TryOn Inference [ComfyUI-Flux-TryOff]",
"TryOnOffRunNode": "Run TryOn or TryOff Inference [ComfyUI-Flux-TryOff]",
}