Update pipeline.py

pipeline.py

@@ -96,7 +96,7 @@ class VidToMePipeline(DiffusionPipeline):
 
     def _build_config(self, video_path, video_prompt, edit_prompt, control_type, 
                   n_timesteps, guidance_scale, negative_prompt, frame_range, 
-                  use_lora, seed, local_merge_ratio, global_merge_ratio):
+                  use_lora, seed, local_merge_ratio, global_merge_ratio, gen_control="none"):
         # Build config using OmegaConf, abstracting as much as possible
         config = OmegaConf.create({
             'sd_version': self.sd_version,  # Default sd_version
@@ -120,7 +120,7 @@ class VidToMePipeline(DiffusionPipeline):
                 'force': False,  # Default, force inversion even if latents exist
             },
             'generation': {
-                'control': "pnp",  # Default to Plug-and-Play for generation control
+                'control': gen_control or "none",  # Default to Plug-and-Play for generation control
                 'pnp_attn_t': 0.5,  # PnP args
                 'pnp_f_t': 0.8,  # PnP args
                 'control_scale': 1.0,  # Scale for ControlNet-like controls