move scheduler timestamps to correct device if tensors

Author: NouamaneTazi

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py

@@ -234,6 +234,8 @@ def __call__(
 
         # set timesteps
         self.scheduler.set_timesteps(num_inference_steps)
+        if isinstance(self.scheduler.timesteps, torch.Tensor):
+            self.scheduler.timesteps = self.scheduler.timesteps.to(self.device)
 
         # if we use LMSDiscreteScheduler, let's make sure latents are multiplied by sigmas
         if isinstance(self.scheduler, LMSDiscreteScheduler):