29 lines
1.3 KiB
Python
29 lines
1.3 KiB
Python
from .nodes.cascades import NODE_CLASS_MAPPINGS as CASCADES_NODE_CLASS_MAPPINGS
|
|
from .nodes.cascades import NODE_DISPLAY_NAME_MAPPINGS as CASCADES_NODE_DISPLAY_NAME_MAPPINGS
|
|
from .nodes.luts import NODE_CLASS_MAPPINGS as LUTS_NODE_CLASS_MAPPINGS
|
|
from .nodes.luts import NODE_DISPLAY_NAME_MAPPINGS as LUTS_NODE_DISPLAY_NAME_MAPPINGS
|
|
from .nodes.waifu2x import NODE_CLASS_MAPPINGS as WAIFU2X_NODE_CLASS_MAPPINGS
|
|
from .nodes.waifu2x import NODE_DISPLAY_NAME_MAPPINGS as WAIFU2X_NODE_DISPLAY_NAME_MAPPINGS
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
# Cascades
|
|
NODE_CLASS_MAPPINGS.update(CASCADES_NODE_CLASS_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(CASCADES_NODE_DISPLAY_NAME_MAPPINGS)
|
|
print(f'\033[93m[Shibiko AI] \033[31mCascades \033[92mLoaded\033[0m')
|
|
|
|
# Luts
|
|
NODE_CLASS_MAPPINGS.update(LUTS_NODE_CLASS_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(LUTS_NODE_DISPLAY_NAME_MAPPINGS)
|
|
print(f'\033[93m[Shibiko AI] \033[31mLuts \033[92mLoaded\033[0m')
|
|
|
|
# Waifu2x
|
|
NODE_CLASS_MAPPINGS.update(WAIFU2X_NODE_CLASS_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(WAIFU2X_NODE_DISPLAY_NAME_MAPPINGS)
|
|
print(f'\033[93m[Shibiko AI] \033[31mWaifu2x \033[92mLoaded\033[0m')
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', 'WEB_DIRECTORY']
|
|
print(f'\033[93m[Shibiko AI] \033[92mLoading Complete!\033[0m')
|