bugfix makeup transfer error

2023-10-30 20:56:51 +08:00
parent 3c61c01ba1
commit e45b89d111
1 changed files with 2 additions and 0 deletions
@@ -369,6 +369,8 @@ class MakeUpTransferPM:
    CATEGORY = "protrait/model"

    def makeup_transfer(self, source_image, makeup_image):
+        source_image = tensor_to_img(source_image)
+        makeup_image = tensor_to_img(makeup_image)
        result = get_pagan_interface().transfer(source_image, makeup_image)
        return (img_to_tensor(result),)