Update app.py

app.py

@@ -17,8 +17,9 @@ login(token=token)
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-video-diffusion-img2vid-xt-1-1")
 #pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
-pipe.enable_xformers_memory_efficient_attention()
 pipe = pipe.to(device)
+pipe.enable_xformers_memory_efficient_attention()
+
 max_64_bit_int = 2**63 - 1
 
 def sample(