debug

Video-P2P/run_videop2p.py

@@ -104,11 +104,11 @@ def main(
     mask_th = (.3, .3)
 
 
-    pretrained_model_path = output_dir
+    pretrained_model_path = pretrained_model_path
     image_path = train_data['video_path']
     prompt = train_data['prompt']
     # prompts = [prompt, ]
-    output_folder = os.path.join(pretrained_model_path, 'results')
+    output_folder = os.path.join(output_dir, 'results')
     if fast:
         save_name_1 = os.path.join(output_folder, 'inversion_fast.gif')
         save_name_2 = os.path.join(output_folder, '{}_fast.gif'.format(save_name))

trainer.py

@@ -145,9 +145,9 @@ class Trainer:
         with open(config_path, 'w') as f:
             OmegaConf.save(config, f)
 
-        command = f'accelerate launch Video-P2P/run_tuning.py --config {config_path}'
-        subprocess.run(shlex.split(command))
-        torch.cuda.empty_cache()
+        # command = f'accelerate launch Video-P2P/run_tuning.py --config {config_path}'
+        # subprocess.run(shlex.split(command))
+        # torch.cuda.empty_cache()
         command = f'python Video-P2P/run_videop2p.py --config {config_path}'
         subprocess.run(shlex.split(command))
         save_model_card(save_dir=output_dir,