File size: 626 Bytes
87614e3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import os

import folder_paths

from .HumanParserLIPCustomNode import HumanParserLIPCustomNode
from .HumanParserATRCustomNode import HumanParserATRCustomNode
from .HumanParserPascalCustomNode import HumanParserPascalCustomNode

# Register the SCHP model folder with ComfyUI so folder_paths can resolve it.
folder_paths.add_model_folder_path("schp", os.path.join(folder_paths.models_dir, "schp"))

NODE_CLASS_MAPPINGS = {
  "Cozy Human Parser LIP" : HumanParserLIPCustomNode,
  "Cozy Human Parser ATR" : HumanParserATRCustomNode,
  "Cozy Human Parser Pascal" : HumanParserPascalCustomNode,
}

__all__ = ['NODE_CLASS_MAPPINGS']