34 lines
1.2 KiB
Python
34 lines
1.2 KiB
Python
# from .mine_nodes import *
|
|
from .nodes.flux_nodes import *
|
|
from .nodes.image_nodes import *
|
|
from .nodes.sampling_nodes import *
|
|
from .nodes.lora_nodes import *
|
|
|
|
# A dictionary that contains all nodes you want to export with their names
|
|
# NOTE: names should be globally unique
|
|
NODE_CLASS_MAPPINGS = {
|
|
"CFE Aspect Ratio" : CFE_Aspect_Ratio,
|
|
"CFE Flux In Pipe" : CFE_Flux_In_Pipe,
|
|
"CFE Flux Out Pipe" : CFE_Flux_Out_Pipe,
|
|
"CFE FLUX Guidance" : CFE_Flux_Guidance,
|
|
"CFE FLUX Sampler (Pipe)" : CFE_FLUX_Pipe_Sampler,
|
|
"CFE FLUX Sampler" : CFE_FLUX_Sampler,
|
|
"CFE Lora Params" : CFE_Lora_Params,
|
|
"CFE Scheduler" : CFE_Scheduler,
|
|
"CFE Sigma Sampler" : CFE_Sigma_Scheduler,
|
|
"CFE Sigma Sampler Strings" : CFE_Sigma_Sampler_Strings,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"CFE_Aspect_Ratio":"CFE Aspect Ratio",
|
|
"CFE_Flux_In_Pipe":"CFE Flux In Pipe",
|
|
"CFE_Flux_Out_Pipe":"CFE Flux Out Pipe",
|
|
"CFE_Flux_Guidance":"CFE Flux Guidance",
|
|
"CFE_FLUX_Pipe_Sampler":"CFE Flux Sampler (Pipe)",
|
|
"CFE_FLUX_Sampler":"CFE Flux Sampler",
|
|
"CFE_Lora_Params":"CFE Lora Params",
|
|
"CFE_Scheduler":"CFE Scheduler",
|
|
"CFE_Sigma_Scheduler":"CFE Sigma Sampler",
|
|
"CFE_Sigma_Sampler_Strings":"CFE Sigma Sampler Strings",
|
|
}
|