ShaoTengLiu commited on
Commit
0fd2709
1 Parent(s): 1a3a206
Files changed (1) hide show
  1. trainer.py +2 -2
trainer.py CHANGED
@@ -137,8 +137,8 @@ class Trainer:
137
  OmegaConf.save(config, f)
138
 
139
  # command = f'accelerate launch Tune-A-Video/train_tuneavideo.py --config {config_path}'
140
- command = f'accelerate launch Video-P2P/run_tuning.py --config {config_path}'
141
- # command = f'accelerate launch Tune-A-Video-debug/train_tuneavideo.py --config {config_path}'
142
  subprocess.run(shlex.split(command))
143
  save_model_card(save_dir=output_dir,
144
  base_model=base_model,
 
137
  OmegaConf.save(config, f)
138
 
139
  # command = f'accelerate launch Tune-A-Video/train_tuneavideo.py --config {config_path}'
140
+ # command = f'accelerate launch Video-P2P/run_tuning.py --config {config_path}'
141
+ command = f'accelerate launch Tune-A-Video-debug/train_tuneavideo.py --config {config_path}'
142
  subprocess.run(shlex.split(command))
143
  save_model_card(save_dir=output_dir,
144
  base_model=base_model,