diff --git a/__init__.py b/__init__.py index cd34471..cbb1dc9 100644 --- a/__init__.py +++ b/__init__.py @@ -58,7 +58,7 @@ for url, filename in zip(urls, filenames): NODE_CLASS_MAPPINGS = { "RetainFace": RetainFace, - "FaceFusion": FaceFusion, + "FaceFusionPM": FaceFusionPM, "RatioMerge2Image": RatioMerge2Image, "MaskMerge2Image": MaskMerge2Image, "ReplaceBoxImg": ReplaceBoxImg, @@ -75,7 +75,7 @@ NODE_CLASS_MAPPINGS = { } NODE_DISPLAY_NAME_MAPPINGS = { "RetainFace": "RetainFace PM", - "FaceFusion": "FaceFusion PM", + "FaceFusionPM": "FaceFusion PM", "RatioMerge2Image": "RatioMerge2Image PM", "MaskMerge2Image": "MaskMerge2Image PM", "ReplaceBoxImg": "ReplaceBoxImg PM", diff --git a/node.py b/node.py index f2f59f1..f592d35 100644 --- a/node.py +++ b/node.py @@ -35,7 +35,7 @@ class RetainFace: crop_image = image.crop(retinaface_boxes[0]) return (img_to_tensor(crop_image), retinaface_tensor, retinaface_boxes[0]) -class FaceFusion: +class FaceFusionPM: def __init__(self): self.image_face_fusion = pipeline(Tasks.image_face_fusion, model='damo/cv_unet-image-face-fusion_damo', model_revision='v1.3') diff --git a/workflows/easyphoto.json b/workflows/easyphoto.json index 3003c0d..7cd324c 100644 --- a/workflows/easyphoto.json +++ b/workflows/easyphoto.json @@ -2635,7 +2635,7 @@ }, { "id": 133, - "type": "FaceFusion", + "type": "FaceFusionPM", "pos": [ 4792, 919 @@ -2672,7 +2672,7 @@ } ], "properties": { - "Node name for S&R": "FaceFusion" + "Node name for S&R": "FaceFusionPM" }, "color": "#432", "bgcolor": "#653" @@ -3258,7 +3258,7 @@ }, { "id": 34, - "type": "FaceFusion", + "type": "FaceFusionPM", "pos": [ 2059, 919 @@ -3295,7 +3295,7 @@ } ], "properties": { - "Node name for S&R": "FaceFusion" + "Node name for S&R": "FaceFusionPM" }, "color": "#432", "bgcolor": "#653"