__init__.py

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