nodes/Cyber-BlackCat-ComfyUI_Auto_Caption
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update auto_caption.py

Browse Source
This commit is contained in:
Cyber-BCat
2024-09-20 10:56:36 +08:00
committed by GitHub
parent aed08e2636
commit 888c985124
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
auto_caption.py
+5 -5
View File
@@ -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)
return (images, masks, image_path_list)