Spaces:
Runtime error
Runtime error
ShaoTengLiu
commited on
Commit
•
18f46c2
1
Parent(s):
7c1d69c
debug
Browse files- trainer.py +3 -2
trainer.py
CHANGED
@@ -18,6 +18,7 @@ from app_upload import ModelUploader
|
|
18 |
from utils import save_model_card
|
19 |
|
20 |
sys.path.append('Tune-A-Video')
|
|
|
21 |
|
22 |
URL_TO_JOIN_MODEL_LIBRARY_ORG = 'https://huggingface.co/organizations/video-p2p-library/share/pZwQaStCpdmMCGLURsMhMkEpvIlsdMdnkk'
|
23 |
ORIGINAL_SPACE_ID = 'video-p2p-library/Video-P2P-Demo'
|
@@ -136,8 +137,8 @@ class Trainer:
|
|
136 |
OmegaConf.save(config, f)
|
137 |
|
138 |
# command = f'accelerate launch Tune-A-Video/train_tuneavideo.py --config {config_path}'
|
139 |
-
|
140 |
-
command = f'accelerate launch Tune-A-Video-debug/train_tuneavideo.py --config {config_path}'
|
141 |
subprocess.run(shlex.split(command))
|
142 |
save_model_card(save_dir=output_dir,
|
143 |
base_model=base_model,
|
|
|
18 |
from utils import save_model_card
|
19 |
|
20 |
sys.path.append('Tune-A-Video')
|
21 |
+
sys.path.append('Video-P2P')
|
22 |
|
23 |
URL_TO_JOIN_MODEL_LIBRARY_ORG = 'https://huggingface.co/organizations/video-p2p-library/share/pZwQaStCpdmMCGLURsMhMkEpvIlsdMdnkk'
|
24 |
ORIGINAL_SPACE_ID = 'video-p2p-library/Video-P2P-Demo'
|
|
|
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,
|