From 635509f00b7cc5519c8bf3b2f06d734c36f81bf8 Mon Sep 17 00:00:00 2001 From: bmaltais Date: Sat, 4 Mar 2023 08:04:38 -0500 Subject: [PATCH] Fix caption by git --- library/git_caption_gui.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/library/git_caption_gui.py b/library/git_caption_gui.py index c65fb3c..3cc69d4 100644 --- a/library/git_caption_gui.py +++ b/library/git_caption_gui.py @@ -25,7 +25,7 @@ def caption_images( return print(f'GIT captioning files in {train_data_dir}...') - run_cmd = f'.\\venv\\Scripts\\python.exe "finetune/make_captions.py"' + run_cmd = f'.\\venv\\Scripts\\python.exe "finetune/make_captions_by_git.py"' if not model_id == '': run_cmd += f' --model_id="{model_id}"' run_cmd += f' --batch_size="{int(batch_size)}"'