From beb1587f715f2e470131451a56dd2b07d73fecff Mon Sep 17 00:00:00 2001 From: iyume Date: Sat, 24 Feb 2024 03:29:59 +0800 Subject: [PATCH] add cfg, steps --- __init__.py | 17 ++++++++++++++--- 1 file changed, 14 insertions(+), 3 deletions(-) diff --git a/__init__.py b/__init__.py index 025f019..cb0def4 100644 --- a/__init__.py +++ b/__init__.py @@ -46,6 +46,17 @@ class OOTDGenerate: # "keypoints": ("POSE_KEYPOINT",), # "category": ("STRING", ["upperbody", "lowerbody", "dress"]), "seed": ("INT", {"default": 0, "min": 0, "max": 0xFFFFFFFFFFFFFFFF}), + "steps": ("INT", {"default": 20, "min": 1, "max": 10000}), + "cfg": ( + "FLOAT", + { + "default": 2.0, + "min": 0.0, + "max": 14.0, + "step": 0.1, + "round": 0.01, + }, + ), } } @@ -55,7 +66,7 @@ class OOTDGenerate: CATEGORY = "OOTD" - def generate(self, pipe, cloth_image, model_image, seed): + def generate(self, pipe, cloth_image, model_image, seed, steps, cfg): category = "upperbody" # if model_image.shape != (1, 1024, 768, 3) or ( # cloth_image.shape != (1, 1024, 768, 3) @@ -100,8 +111,8 @@ class OOTDGenerate: mask=mask, image_ori=model_image, num_samples=1, - num_steps=20, - image_scale=1, + num_steps=steps, + image_scale=cfg, seed=seed, )