update video-p2p

requirements.txt

@@ -17,3 +17,5 @@ torchvision==0.14.1
 transformers==4.26.0
 triton==2.0.0.dev20221202
 xformers==0.0.16
+opencv-python
+ipywidgets

trainer.py

@@ -136,6 +136,10 @@ class Trainer:
         config.prompts = [training_prompt, validation_prompt]
         config.blend_word = [blend_word_1, blend_word_2]
         config.eq_params = {"words":eq_params_1, "values":int(eq_params_2)}
+        if len(validation_prompt) == len(training_prompt):
+            config.is_word_swap = True
+        else:
+            config.is_word_swap = False
 
         config_path = output_dir / 'config.yaml'
         with open(config_path, 'w') as f: