Spaces:
Runtime error
Runtime error
ShaoTengLiu
commited on
Commit
·
1276ecf
1
Parent(s):
23ca4f1
debug
Browse files- trainer.py +1 -0
trainer.py
CHANGED
|
@@ -147,6 +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_videop2p.py --config {config_path}'
|
| 151 |
subprocess.run(shlex.split(command))
|
| 152 |
save_model_card(save_dir=output_dir,
|
|
|
|
| 147 |
|
| 148 |
command = f'accelerate launch Video-P2P/run_tuning.py --config {config_path}'
|
| 149 |
subprocess.run(shlex.split(command))
|
| 150 |
+
torch.cuda.empty_cache()
|
| 151 |
command = f'python Video-P2P/run_videop2p.py --config {config_path}'
|
| 152 |
subprocess.run(shlex.split(command))
|
| 153 |
save_model_card(save_dir=output_dir,
|