File size: 2,086 Bytes
1900386 |
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 50 |
# 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"
}
|