Files
2024-08-08 22:58:23 +08:00

28 lines
946 B
Python

from .nodes.live_portrait import LivePortraitNode,FaceCropInfo,Retargeting,LivePortraitVideoNode
from .nodes.expression_editor import ExpressionEditor,ExpressionVideoNode,ExpressionVideo2VideoNode
NODE_CLASS_MAPPINGS = {
"LivePortraitNode": LivePortraitNode,
"LivePortraitVideoNode":LivePortraitVideoNode,
"FaceCropInfo":FaceCropInfo,
"Retargeting":Retargeting,
"ExpressionEditor_":ExpressionEditor,
"ExpressionVideoNode":ExpressionVideoNode,
"ExpressionVideo2VideoNode":ExpressionVideo2VideoNode
}
# dict = { "key":value }
NODE_DISPLAY_NAME_MAPPINGS = {
"LivePortraitNode":"Live Portrait",
"LivePortraitVideoNode":"Live Portrait for Video",
"FaceCropInfo":"Face Crop Info",
"Retargeting":"Retargeting",
"ExpressionEditor_":"Expression Editor",
"ExpressionVideoNode":"Expression Video",
"ExpressionVideo2VideoNode":"Expression Video 2 Video"
}
# web ui的节点功能
WEB_DIRECTORY = "./web"