From 5648fdf34134e75fa25b1cec601821367bcc4f96 Mon Sep 17 00:00:00 2001 From: Brandon Rising Date: Tue, 28 May 2024 14:57:04 -0400 Subject: [PATCH] Allow --output flag to be passed to autocaptioning script to overwrite default output.jsonl --- .gitignore | 1 + .../scripts/_experimental/auto_caption/auto_caption_images.py | 2 -- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/.gitignore b/.gitignore index 52c2e482..76b01973 100644 --- a/.gitignore +++ b/.gitignore @@ -169,3 +169,4 @@ cython_debug/ # and can be added to the global gitignore or merged into this file. For a more nuclear # option (not recommended) you can uncomment the following to ignore the entire idea folder. #.idea/ +.aider* diff --git a/src/invoke_training/scripts/_experimental/auto_caption/auto_caption_images.py b/src/invoke_training/scripts/_experimental/auto_caption/auto_caption_images.py index 374e6083..301323ff 100644 --- a/src/invoke_training/scripts/_experimental/auto_caption/auto_caption_images.py +++ b/src/invoke_training/scripts/_experimental/auto_caption/auto_caption_images.py @@ -108,5 +108,3 @@ def main(image_dir: str, prompt: str, use_cpu: bool, batch_size: int, output_pat args = parser.parse_args() main(args.dir, args.prompt, args.cpu, args.batch_size, args.output) - - main(args.dir, args.prompt, args.cpu, args.batch_size)