update

Video-P2P/run_videop2p.py

@@ -87,7 +87,7 @@ def main(
     cross_replace_steps: float = 0.2,
     self_replace_steps: float = 0.5,
     video_len: int = 8,
-    fast: bool = False,
+    fast: bool = True,
     mixed_precision_p2p: str = 'fp32',
 ):
 
@@ -695,7 +695,8 @@ def main(
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
     parser.add_argument("--config", type=str, default="./configs/tuneavideo.yaml")
-    parser.add_argument("--fast", action='store_true')
+    # parser.add_argument("--fast", action='store_true')
     args = parser.parse_args()
 
-    main(**OmegaConf.load(args.config), fast=args.fast)
+    # main(**OmegaConf.load(args.config), fast=args.fast)
+    main(**OmegaConf.load(args.config))

trainer.py

@@ -147,7 +147,7 @@ class Trainer:
 
         command = f'accelerate launch Video-P2P/run_tuning.py --config {config_path}'
         subprocess.run(shlex.split(command))
-        command = f'python Video-P2P/run_videop2p.py --config {config_path} --fast'
+        command = f'python Video-P2P/run_videop2p.py --config {config_path}'
         subprocess.run(shlex.split(command))
         save_model_card(save_dir=output_dir,
                         base_model=base_model,