Create run_vidtome.py

run_vidtome.py

@@ -0,0 +1,21 @@
+from invert import Inverter
+from generate import Generator
+from utils import load_config, init_model, seed_everything, get_frame_ids
+
+if __name__ == "__main__":
+    config = load_config()
+    pipe, scheduler, model_key = init_model(
+        config.device, config.sd_version, config.model_key, config.generation.control, config.float_precision)
+    config.model_key = model_key
+    seed_everything(config.seed)
+    
+    print("Start inversion!")
+    inversion = Inverter(pipe, scheduler, config)
+    inversion(config.input_path, config.inversion.save_path)
+
+    print("Start generation!")
+    generator = Generator(pipe, scheduler, config)
+    frame_ids = get_frame_ids(
+        config.generation.frame_range, config.generation.frame_ids)
+    generator(config.input_path, config.generation.latents_path,
+              config.generation.output_path, frame_ids=frame_ids)