modify FaceFusion name to FaceFusionPM for fix conflict with other plugin

This commit is contained in:
toto
2023-10-18 15:03:20 +08:00
parent 0db9951416
commit 086a2cae72
3 changed files with 7 additions and 7 deletions
+2 -2
View File
@@ -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",
+1 -1
View File
@@ -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')
+4 -4
View File
@@ -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"