Allow --output flag to be passed to autocaptioning script to overwrit…

…e default output.jsonl
invoke-ai · May 28, 2024 · 5648fdf · 5648fdf
1 parent 228696b
commit 5648fdf
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git 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
@@ -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)