From 02f414b637bf54943e676213d3f9e2eeb9ede232 Mon Sep 17 00:00:00 2001
From: wailovet <wailovet@163.com>
Date: Fri, 28 Jun 2024 01:01:24 +0800
Subject: [PATCH] 'update'

---
 hook_HYDiT_main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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")