fix bug

Video-P2P/tuneavideo/data/dataset.py

@@ -34,7 +34,7 @@ class TuneAVideoDataset(Dataset):
             for file in sorted(os.listdir(self.video_path), key=lambda x: int(x[:-4])):
                 if file.endswith('jpg'):
                     self.images.append(np.asarray(Image.open(os.path.join(self.video_path, file)).convert('RGB').resize((self.width, self.height))))
-        self.images = np.stack(self.images)
+            self.images = np.stack(self.images)
 
     def __len__(self):
         return 1

trainer.py

@@ -137,9 +137,9 @@ 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 Video-P2P/run_tuning.py --config {config_path}'
         # command = f'accelerate launch Video-P2P/train_tuneavideo.py --config {config_path}'
-        command = f'accelerate launch Tune-A-Video-debug/train_tuneavideo.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,