Update skyreels_v2_infer/pipelines/text2video_pipeline.py

skyreels_v2_infer/pipelines/text2video_pipeline.py

@@ -83,7 +83,7 @@ class Text2VideoPipeline:
 
         # evaluation mode
         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