update

README.md

@@ -1,12 +1,12 @@
 ---
-title: Tune-A-Video Training UI
+title: Video-P2P UI
 emoji: ⚡
 colorFrom: red
 colorTo: purple
 sdk: docker
 pinned: false
 license: mit
-duplicated_from: lora-library/LoRA-DreamBooth-Training-UI
+duplicated_from: Tune-A-Video-library/Tune-A-Video-Training-UI
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

inference.py

@@ -13,6 +13,7 @@ from diffusers.utils.import_utils import is_xformers_available
 from einops import rearrange
 from huggingface_hub import ModelCard
 
+sys.path.append('Video-P2P')
 sys.path.append('Tune-A-Video')
 
 from tuneavideo.models.unet import UNet3DConditionModel

trainer.py

@@ -137,8 +137,8 @@ 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/train_tuneavideo.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}'
         subprocess.run(shlex.split(command))
         save_model_card(save_dir=output_dir,