| 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'] | |