From 7feecc57a10f4b15fc32c0c54e0a6fba5e653606 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Sun, 25 Aug 2024 17:04:06 +0300 Subject: [PATCH] Add prefix/suffix to folder caption node --- nodes.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/nodes.py b/nodes.py index e19ac57..0dc1061 100644 --- a/nodes.py +++ b/nodes.py @@ -222,6 +222,8 @@ class OneVisionCaptionFolder: "temperature": ("FLOAT", {"default": 0.2, "min": 0.0, "max": 1.0, "step": 0.01}), "seed": ("INT", {"default": 1, "min": 1, "max": 0xffffffffffffffff}), "max_image_size": ("INT", {"default": 1024, "min": 256, "max": 8192}), + "prefix": ("STRING", {"default": "",}), + "suffix": ("STRING", {"default": "",}), }, } @@ -230,7 +232,7 @@ class OneVisionCaptionFolder: OUTPUT_IS_LIST = (True,) CATEGORY = "LLaVA-OneVision" - def caption(self, folder_path, llava_model, prompt, max_tokens, keep_model_loaded, temperature, seed, max_image_size): + def caption(self, folder_path, llava_model, prompt, max_tokens, keep_model_loaded, temperature, seed, max_image_size, prefix, suffix): from PIL import Image image_files = [] for filename in os.listdir(folder_path): @@ -265,6 +267,7 @@ class OneVisionCaptionFolder: temperature=temperature, seed=seed) + result = prefix + result + suffix results_list.append(result) base_filename = os.path.splitext(img_path)[0]