Spaces:
Runtime error
Runtime error
ShaoTengLiu
commited on
Commit
·
2b2d19c
1
Parent(s):
1eeee01
fix a bug
Browse files- 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/
|
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,
|