From 6a2d356ed75c9a3a0880d20674e5bfa9083771ed Mon Sep 17 00:00:00 2001 From: toto Date: Sat, 16 Dec 2023 21:36:21 +0800 Subject: [PATCH] add some pack --- portrait/nodes.py | 3 ++- requirements.txt | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/portrait/nodes.py b/portrait/nodes.py index 8c5db20..b004fb5 100644 --- a/portrait/nodes.py +++ b/portrait/nodes.py @@ -295,7 +295,7 @@ class ImageScaleShortPM: return {"required": { "image": ("IMAGE",), "size": ("INT", {"default": 512, "min": 0, "max": 2048, "step": 1}), - "crop_face": ("BOOLEAN", {"default": False, "label_on": "enabled", "label_off": "disabled"}), + "crop_face": ("BOOLEAN", {"default": False}), }} RETURN_TYPES = ("IMAGE",) @@ -303,6 +303,7 @@ class ImageScaleShortPM: CATEGORY = "protrait/other" + def image_scale_short(self, image, size, crop_face): input_image = tensor_to_img(image) short_side = min(input_image.width, input_image.height) diff --git a/requirements.txt b/requirements.txt index 1cb78dd..6eb9b26 100644 --- a/requirements.txt +++ b/requirements.txt @@ -9,4 +9,5 @@ scikit-image matplotlib insightface diffusers==0.18.2 +sentencepiece mmcv_full