From 888c985124ba2381781283983db2d950f6d014a3 Mon Sep 17 00:00:00 2001 From: Cyber-BCat Date: Fri, 20 Sep 2024 10:56:36 +0800 Subject: [PATCH] Update auto_caption.py --- auto_caption.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/auto_caption.py b/auto_caption.py index c16539b..6e52f1c 100644 --- a/auto_caption.py +++ b/auto_caption.py @@ -61,7 +61,7 @@ class Joy_Model_load: } } - CATEGORY = "AutoCaption" + CATEGORY = "Auto Caption" RETURN_TYPES = ("JoyModel",) FUNCTION = "gen" @@ -137,7 +137,7 @@ class Auto_Caption: } } - CATEGORY = "AutoCaption" + CATEGORY = "Auto Caption" RETURN_TYPES = ("STRING",) FUNCTION = "gen" def gen(self,JoyModel,image,prompt,max_new_tokens,temperature,cache): @@ -219,7 +219,7 @@ class LoadImagesRezise: RETURN_NAMES = ("image", "mask", "count", "image_path",) FUNCTION = "load_images" - CATEGORY = "AutoCaption" + CATEGORY = "Auto Caption" def load_images(self, folder, image_load_cap, start_index): if not os.path.isdir(folder): @@ -330,7 +330,7 @@ class LoadManyImages: FUNCTION = "load_images" - CATEGORY = "AutoCaption" + CATEGORY = "Auto Caption" def load_images(self, folder, image_load_cap, start_index): if not os.path.isdir(folder): @@ -391,4 +391,4 @@ class LoadManyImages: image_path_list.append(image_path) image_count += 1 - return (images, masks, image_path_list) \ No newline at end of file + return (images, masks, image_path_list)