Update skyreels_v2_infer/pipelines/image2video_pipeline.py

skyreels_v2_infer/pipelines/image2video_pipeline.py

@@ -117,7 +117,7 @@ class Image2VideoPipeline:
         )
 
         self.transformer.to(self.device)
-        with torch.cuda.amp.autocast(dtype=self.transformer.dtype), torch.no_grad():
+        with torch.amp.autocast("cuda", dtype=self.transformer.dtype), torch.no_grad():
             self.scheduler.set_timesteps(num_inference_steps, device=self.device, shift=shift)
             timesteps = self.scheduler.timesteps