debug

Video-P2P/configs/man-surfing-tune.yaml

@@ -4,7 +4,7 @@ output_dir: "./outputs/man-surfing"
 train_data:
   video_path: "data/man-surfing.mp4"
   prompt: "a man is surfing"
-  n_sample_frames: 8
+  n_sample_frames: 24
   width: 512
   height: 512
   sample_start_idx: 0
@@ -13,7 +13,7 @@ train_data:
 validation_data:
   prompts:
     - "a panda is surfing"
-  video_length: 8
+  video_length: 24
   width: 512
   height: 512
   num_inference_steps: 50

trainer.py

@@ -105,8 +105,8 @@ class Trainer:
                 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('Tune-A-Video-debug/configs/man-surfing.yaml')
+        config = OmegaConf.load('Video-P2P/configs/man-surfing-tune.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