modify FaceFusion name to FaceFusionPM for fix conflict with other plugin
This commit is contained in:
+2
-2
@@ -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",
|
||||
|
||||
@@ -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')
|
||||
|
||||
@@ -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"
|
||||
|
||||
Reference in New Issue
Block a user