diff --git a/hook_HYDiT_main.py b/hook_HYDiT_main.py
index 35fe902..af85446 100644
--- a/hook_HYDiT_main.py
+++ b/hook_HYDiT_main.py
@@ -784,7 +784,7 @@ def easy_sample_images(
                                            requires_safety_checker=False,
                                            embedder_t5=embedder_t5,
                                            )
-        pipeline._execution_device = "cuda"
+        
         pipeline.to("cuda")
 
         style = torch.as_tensor([0, 0] * batch_size, device="cuda")