47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
import os
|
|
|
|
from .datastasher import CsvWriterNode
|
|
from .dataloader import ImmatureImageDataLoader
|
|
from .imagecounter import ImmatureImageCounter
|
|
from .cached_checkpoint import CachedCheckpoint
|
|
from .escapedfind import BracketEscaper
|
|
from .girlselector import RandomCharacterSelector
|
|
from .schedulerselector import *
|
|
from .advancedksampler import *
|
|
from .danbooru import Ranbooru
|
|
|
|
|
|
# A dictionary that contains all nodes you want to export with their names
|
|
NODE_CLASS_MAPPINGS = {
|
|
"CsvWriterNode": CsvWriterNode,
|
|
"ImmatureImageDataLoader": ImmatureImageDataLoader,
|
|
"ImmatureImageCounter": ImmatureImageCounter,
|
|
"CachedCheckpoint": CachedCheckpoint,
|
|
"BracketEscaper": BracketEscaper,
|
|
"RandomCharacterSelector": RandomCharacterSelector,
|
|
"RapidSchedulerSelector": RapidSchedulerSelector,
|
|
"RapidSchedulerCombo": RapidSchedulerCombo,
|
|
"Ranbooru": Ranbooru,
|
|
"TwoModelAdvancedKsampler": TwoModelAdvancedKsampler,
|
|
}
|
|
|
|
# A dictionary that contains the friendly/humanly readable titles for the nodes
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"CsvWriterNode": "CSV Writer",
|
|
"ImmatureImageDataLoader": "Immature Image Data Loader",
|
|
"ImmatureImageCounter": "Immature Image Counter",
|
|
"CachedCheckpoint": "Cached Checkpoint",
|
|
"BracketEscaper": "Bracket Escaper",
|
|
"RapidSchedulerSelector": "Scheduler Selector",
|
|
"RapidSchedulerCombo": "Rapid Scheduler Combo",
|
|
"Ranbooru": "Danbooru",
|
|
"TwoModelAdvancedKsampler": "TwoModelAdvancedKsampler"
|
|
}
|
|
|
|
__all__ = NODE_CLASS_MAPPINGS
|
|
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|
|
|
|
|