nodes/kijai-ComfyUI-LLaVA-OneVision
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add prefix/suffix to folder caption node

Browse Source
This commit is contained in:
kijai
2024-08-25 17:04:06 +03:00
parent 87e3784d14
commit 7feecc57a1
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+4 -1
View File
@@ -222,6 +222,8 @@ class OneVisionCaptionFolder:
"temperature": ("FLOAT", {"default": 0.2, "min": 0.0, "max": 1.0, "step": 0.01}), "temperature": ("FLOAT", {"default": 0.2, "min": 0.0, "max": 1.0, "step": 0.01}),
"seed": ("INT", {"default": 1, "min": 1, "max": 0xffffffffffffffff}), "seed": ("INT", {"default": 1, "min": 1, "max": 0xffffffffffffffff}),
"max_image_size": ("INT", {"default": 1024, "min": 256, "max": 8192}), "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,) OUTPUT_IS_LIST = (True,)
CATEGORY = "LLaVA-OneVision" 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 from PIL import Image
image_files = [] image_files = []
for filename in os.listdir(folder_path): for filename in os.listdir(folder_path):
@@ -265,6 +267,7 @@ class OneVisionCaptionFolder:
temperature=temperature, temperature=temperature,
seed=seed) seed=seed)
result = prefix + result + suffix
results_list.append(result) results_list.append(result)
base_filename = os.path.splitext(img_path)[0] base_filename = os.path.splitext(img_path)[0]