comfyui-liveportrait/__init__.py at main · MixLabPro/comfyui-liveportrait · GitHub

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
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"