ShaoTengLiu commited on
Commit
2b2d19c
1 Parent(s): 1eeee01
Files changed (1) hide show
  1. trainer.py +1 -1
trainer.py CHANGED
@@ -133,7 +133,7 @@ class Trainer:
133
  with open(config_path, 'w') as f:
134
  OmegaConf.save(config, f)
135
 
136
- command = f'accelerate launch Video-P2P-Demo/train_tuneavideo.py --config {config_path}'
137
  subprocess.run(shlex.split(command))
138
  save_model_card(save_dir=output_dir,
139
  base_model=base_model,
 
133
  with open(config_path, 'w') as f:
134
  OmegaConf.save(config, f)
135
 
136
+ command = f'accelerate launch Video-P2P-Demo/run_tuning.py --config {config_path}'
137
  subprocess.run(shlex.split(command))
138
  save_model_card(save_dir=output_dir,
139
  base_model=base_model,