Fix caption by git
This commit is contained in:
parent
4c1448be72
commit
635509f00b
@ -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)}"'
|
||||
|
Loading…
Reference in New Issue
Block a user