Fix caption by git

This commit is contained in:
bmaltais 2023-03-04 08:04:38 -05:00
parent 4c1448be72
commit 635509f00b

View File

@ -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)}"'