ApacheOne commited on
Commit
87614e3
·
verified ·
1 Parent(s): d38f111

Update filefixes/__init__.py

Browse files
Files changed (1) hide show
  1. filefixes/__init__.py +18 -0
filefixes/__init__.py CHANGED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+
3
+ import folder_paths
4
+
5
+ from .HumanParserLIPCustomNode import HumanParserLIPCustomNode
6
+ from .HumanParserATRCustomNode import HumanParserATRCustomNode
7
+ from .HumanParserPascalCustomNode import HumanParserPascalCustomNode
8
+
9
+ # Register the SCHP model folder with ComfyUI so folder_paths can resolve it.
10
+ folder_paths.add_model_folder_path("schp", os.path.join(folder_paths.models_dir, "schp"))
11
+
12
+ NODE_CLASS_MAPPINGS = {
13
+ "Cozy Human Parser LIP" : HumanParserLIPCustomNode,
14
+ "Cozy Human Parser ATR" : HumanParserATRCustomNode,
15
+ "Cozy Human Parser Pascal" : HumanParserPascalCustomNode,
16
+ }
17
+
18
+ __all__ = ['NODE_CLASS_MAPPINGS']