Revert torch_device to cpu for QwenImageEdit tests that run out of memory on MPS

Author: Sunhill666

tests/pipelines/qwenimage/test_qwenimage_edit.py

@@ -245,16 +245,16 @@ def test_encode_prompt_works_in_isolation(self, extra_required_param_value_dict=
     def test_true_cfg_without_negative_prompt_embeds_mask(self):
         components = self.get_dummy_components()
         pipe = self.pipeline_class(**components)
-        pipe.to(torch_device)
+        pipe.to("cpu")
         pipe.set_progress_bar_config(disable=None)
 
-        inputs = self.get_dummy_inputs(torch_device)
+        inputs = self.get_dummy_inputs("cpu")
         prompt = inputs.pop("prompt")
 
         prompt_embeds, prompt_embeds_mask = pipe.encode_prompt(
             prompt=prompt,
             image=inputs.get("image") if "image" in inputs else None,
-            device=torch_device,
+            device="cpu",
             num_images_per_prompt=1,
             max_sequence_length=inputs.get("max_sequence_length", 16),
         )

tests/pipelines/qwenimage/test_qwenimage_edit_plus.py

@@ -255,16 +255,16 @@ def test_inference_batch_single_identical():
     def test_true_cfg_without_negative_prompt_embeds_mask(self):
         components = self.get_dummy_components()
         pipe = self.pipeline_class(**components)
-        pipe.to(torch_device)
+        pipe.to("cpu")
         pipe.set_progress_bar_config(disable=None)
 
-        inputs = self.get_dummy_inputs(torch_device)
+        inputs = self.get_dummy_inputs("cpu")
         prompt = inputs.pop("prompt")
 
         prompt_embeds, prompt_embeds_mask = pipe.encode_prompt(
             prompt=prompt,
             image=inputs.get("image") if "image" in inputs else None,
-            device=torch_device,
+            device="cpu",
             num_images_per_prompt=1,
             max_sequence_length=inputs.get("max_sequence_length", 16),
         )