fix offload test (#7868)

fix

Co-authored-by: Dhruv Nair <[email protected]>

Author: yiyixuxu

tests/pipelines/test_pipelines_common.py

@@ -1375,7 +1375,7 @@ def test_sequential_cpu_offload_forward_pass(self, expected_max_diff=1e-4):
         output_without_offload = pipe(**inputs)[0]
 
         pipe.enable_sequential_cpu_offload()
-        assert pipe._execution_device.type == pipe._offload_device.type
+        assert pipe._execution_device.type == "cuda"
 
         inputs = self.get_dummy_inputs(generator_device)
         output_with_offload = pipe(**inputs)[0]
@@ -1440,7 +1440,7 @@ def test_model_cpu_offload_forward_pass(self, expected_max_diff=2e-4):
         output_without_offload = pipe(**inputs)[0]
 
         pipe.enable_model_cpu_offload()
-        assert pipe._execution_device.type == pipe._offload_device.type
+        assert pipe._execution_device.type == "cuda"
 
         inputs = self.get_dummy_inputs(generator_device)
         output_with_offload = pipe(**inputs)[0]