torch_device instead of cuda.

Author: sayakpaul

tests/pipelines/controlnet_flux/test_controlnet_flux.py

@@ -210,7 +210,7 @@ def test_canny(self):
             text_encoder_2=None,
             controlnet=controlnet,
             torch_dtype=torch.bfloat16,
-        ).to("cuda")
+        ).to(torch_device)
         pipe.set_progress_bar_config(disable=None)
 
         generator = torch.Generator(device="cpu").manual_seed(0)