ShaoTengLiu commited on
Commit
8bad22b
1 Parent(s): 709f3d4
Files changed (1) hide show
  1. trainer.py +2 -2
trainer.py CHANGED
@@ -138,8 +138,8 @@ class Trainer:
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 Video-P2P/train_tuneavideo.py --config {config_path}'
142
- # command = f'accelerate launch Tune-A-Video-debug/train_tuneavideo.py --config {config_path}'
143
  subprocess.run(shlex.split(command))
144
  save_model_card(save_dir=output_dir,
145
  base_model=base_model,
 
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 Video-P2P/train_tuneavideo.py --config {config_path}'
142
+ command = f'accelerate launch Tune-A-Video-debug/train_tuneavideo.py --config {config_path}'
143
  subprocess.run(shlex.split(command))
144
  save_model_card(save_dir=output_dir,
145
  base_model=base_model,