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)