Update app.py

app.py

@@ -23,7 +23,6 @@ unet = UNet2DConditionModel.from_pretrained(model_id, subfolder="unet", revision
 
 # pipe for TokenFlow
 tokenflow_pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16).to("cuda")
-tokenflow_pipe.enable_xformers_memory_efficient_attention()
 
 def randomize_seed_fn():
     seed = random.randint(0, np.iinfo(np.int32).max)
@@ -131,7 +130,7 @@ def preprocess_and_invert(input_video,
     sd_version = "2.1"
     height: int = 512
     weidth: int = 512
-
+    
     if do_inversion or randomize_seed:
         preprocess_config = {}
         preprocess_config['H'] = height
@@ -231,7 +230,7 @@ def edit_with_pnp(input_video,
             seed = randomize_seed_fn()
     seed_everything(seed)
     
-    
+    tokenflow_pipe.enable_xformers_memory_efficient_attention()
     editor = TokenFlow(config=config,pipe=tokenflow_pipe, frames=frames.value, inverted_latents=inverted_latents.value, zs= zs.value)
     edited_frames = editor.edit_video()