feat: update node category
This commit is contained in:
@@ -34,7 +34,7 @@ class HumanParserATRCustomNode:
|
||||
RETURN_TYPES = ("MASK", "IMAGE")
|
||||
RETURN_NAMES = ("mask", "map")
|
||||
FUNCTION = "run"
|
||||
CATEGORY = "examples"
|
||||
CATEGORY = "CozyMantis"
|
||||
|
||||
def run(self, image, background, hat, hair, sunglasses, upper_clothes, skirt, pants, dress, belt, left_shoe, right_shoe, face, left_leg, right_leg, left_arm, right_arm, bag, scarf):
|
||||
if torch.cuda.is_available():
|
||||
|
||||
@@ -36,7 +36,7 @@ class HumanParserLIPCustomNode:
|
||||
RETURN_TYPES = ("MASK", "IMAGE")
|
||||
RETURN_NAMES = ("mask", "map")
|
||||
FUNCTION = "run"
|
||||
CATEGORY = "examples"
|
||||
CATEGORY = "CozyMantis"
|
||||
|
||||
def run(self, image, background, hat, hair, glove, sunglasses, upper_clothes, dress, coat, socks, pants, jumpsuits, scarf, skirt, face, left_arm, right_arm, left_leg, right_leg, left_shoe, right_shoe):
|
||||
if torch.cuda.is_available():
|
||||
|
||||
@@ -23,7 +23,7 @@ class HumanParserPascalCustomNode:
|
||||
RETURN_TYPES = ("MASK", "IMAGE")
|
||||
RETURN_NAMES = ("mask", "map")
|
||||
FUNCTION = "run"
|
||||
CATEGORY = "examples"
|
||||
CATEGORY = "CozyMantis"
|
||||
|
||||
def run(self, image, background, head, torso, upper_arms, lower_arms, upper_legs, lower_legs):
|
||||
if torch.cuda.is_available():
|
||||
|
||||
Reference in New Issue
Block a user