Add prefix/suffix to folder caption node

This commit is contained in:
kijai
2024-08-25 17:04:06 +03:00
parent 87e3784d14
commit 7feecc57a1
+4 -1
View File
@@ -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]