ShaoTengLiu commited on
Commit
1cb73da
1 Parent(s): 44fa1db
Files changed (1) hide show
  1. trainer.py +1 -1
trainer.py CHANGED
@@ -147,7 +147,7 @@ class Trainer:
147
 
148
  command = f'accelerate launch Video-P2P/run_tuning.py --config {config_path}'
149
  subprocess.run(shlex.split(command))
150
- command = f'python Video-P2P/run_tuning.py --config {config_path} --fast'
151
  subprocess.run(shlex.split(command))
152
  save_model_card(save_dir=output_dir,
153
  base_model=base_model,
 
147
 
148
  command = f'accelerate launch Video-P2P/run_tuning.py --config {config_path}'
149
  subprocess.run(shlex.split(command))
150
+ command = f'python Video-P2P/run_videop2p.py --config {config_path} --fast'
151
  subprocess.run(shlex.split(command))
152
  save_model_card(save_dir=output_dir,
153
  base_model=base_model,