ApacheOne's picture
Update filefixes/__init__.py
87614e3 verified
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']