debug

trainer.py

@@ -18,6 +18,7 @@ from app_upload import ModelUploader
 from utils import save_model_card
 
 sys.path.append('Tune-A-Video')
+sys.path.append('Video-P2P')
 
 URL_TO_JOIN_MODEL_LIBRARY_ORG = 'https://huggingface.co/organizations/video-p2p-library/share/pZwQaStCpdmMCGLURsMhMkEpvIlsdMdnkk'
 ORIGINAL_SPACE_ID = 'video-p2p-library/Video-P2P-Demo'
@@ -136,8 +137,8 @@ class Trainer:
             OmegaConf.save(config, f)
 
         # command = f'accelerate launch Tune-A-Video/train_tuneavideo.py --config {config_path}'
-        # command = f'accelerate launch Video-P2P/run_tuning.py --config {config_path}'
-        command = f'accelerate launch Tune-A-Video-debug/train_tuneavideo.py --config {config_path}'
+        command = f'accelerate launch Video-P2P/run_tuning.py --config {config_path}'
+        # command = f'accelerate launch Tune-A-Video-debug/train_tuneavideo.py --config {config_path}'
         subprocess.run(shlex.split(command))
         save_model_card(save_dir=output_dir,
                         base_model=base_model,