debug

trainer.py

@@ -103,8 +103,9 @@ class Trainer:
             self.join_model_library_org(
                 self.hf_token if self.hf_token else input_token)
 
+        # config = OmegaConf.load('Tune-A-Video/configs/man-surfing.yaml')
         # config = OmegaConf.load('Video-P2P/configs/man-surfing-tune.yaml')
-        config = OmegaConf.load('Video-P2P/configs/man-surfing.yaml')
+        config = OmegaConf.load('Tune-A-Video-debug/configs/man-surfing.yaml')
         config.pretrained_model_path = self.download_base_model(base_model)
         config.output_dir = output_dir.as_posix()
         config.train_data.video_path = training_video.name  # type: ignore
@@ -134,6 +135,7 @@ class Trainer:
         with open(config_path, 'w') as f:
             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}'
         subprocess.run(shlex.split(command))