-
Notifications
You must be signed in to change notification settings - Fork 33
/
__init__.py
49 lines (43 loc) · 2.04 KB
/
__init__.py
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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
# from .mine_nodes import *
from .utils.file_util import *
import folder_paths
import os
import sys
# comfy_path = os.path.dirname(folder_paths.__file__)
# custom_nodes_path = os.path.join(comfy_path, "custom_nodes")
# import sys
vition_path = node_path("ComfyUI_Seg_VITON")
sys.path.append(vition_path)
# sys.path.append(os.path.join(vition_path,"cldm"))
# sys.path.append(os.path.join(vition_path,"ldm"))
# sys.path.append(os.path.join(vition_path,"ldm","data"))
sys.path.append(os.path.join(vition_path,"ldm","models"))
# sys.path.append(os.path.join(vition_path,"ldm","models","diffusion"))
# sys.path.append(os.path.join(vition_path,"ldm","models","diffusion","dpm_solver"))
# sys.path.append(os.path.join(vition_path,"ldm","modules"))
# sys.path.append(os.path.join(vition_path,"ldm","modules","diffusionmodules"))
# sys.path.append(os.path.join(vition_path,"ldm","modules","distributions"))
# sys.path.append(os.path.join(vition_path,"ldm","modules","encoders"))
# sys.path.append(os.path.join(vition_path,"ldm","modules","image_degradation"))
# sys.path.append(os.path.join(vition_path,"ldm","modules","image_degradation","utils"))
# sys.path.append(os.path.join(vition_path,"ldm","modules","image_encoders"))
# sys.path.append(os.path.join(vition_path,"ldm","modules","midas"))
# sys.path.append(os.path.join(vition_path,"ldm","modules","midas","midas"))
# sys.path.append(os.path.join(vition_path,"utils"))
from .segformer_clothes import *
from .stabel_vition import *
# A dictionary that contains all nodes you want to export with their names
# NOTE: names should be globally unique
NODE_CLASS_MAPPINGS = {
"segformer_clothes":segformer_clothes,
"segformer_agnostic":segformer_agnostic,
"segformer_remove_bg":segformer_remove_bg,
"stabel_vition":stabel_vition
}
# A dictionary that contains the friendly/humanly readable titles for the nodes
NODE_DISPLAY_NAME_MAPPINGS = {
"segformer_clothes":"segformer_clothes",
"segformer_agnostic":"segformer_agnostic",
"segformer_remove_bg":"segformer_remove_bg",
"stabel_vition":"stabel_vition"
}