Mirrors
/
EveryDream2trainer
mirror of https://github.com/victorchall/EveryDream2trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

bug in caption cog if starts_with no there

This commit is contained in:
Victor Hall 2024-03-02 01:58:13 -05:00
parent d098223052
commit 42ad271261
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
caption_cog.py
View File

@ -184,7 +184,7 @@ def main(args):
total_start_time = time.time() total_start_time = time.time()
i_processed = 0 i_processed = 0
starts_with = args.starts_with.strip() starts_with = args.starts_with.strip() if args.starts_with is not None else ""
for i, image_path in enumerate(image_generator(args.image_dir, do_recurse=not args.no_recurse)): for i, image_path in enumerate(image_generator(args.image_dir, do_recurse=not args.no_recurse)):
candidate_caption_path = image_path.replace(os.path.splitext(image_path)[-1], ".txt") candidate_caption_path = image_path.replace(os.path.splitext(image_path)[-1], ".txt")