From 3ff4430e84a9b71aef8bec62999aeabcdf2c1d46 Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Thu, 14 Aug 2025 09:33:44 -0600
Subject: [PATCH] Fix issue with fake text encoder unload

---
 toolkit/unloader.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/toolkit/unloader.py b/toolkit/unloader.py
index 6c45926f..09ce1ba9 100644
--- a/toolkit/unloader.py
+++ b/toolkit/unloader.py
@@ -58,6 +58,6 @@ def unload_text_encoder(model: "BaseModel"):
             model.text_encoder = text_encoder_list
         else:
             # only has a single text encoder
-            model.text_encoder = FakeTextEncoder()
+            model.text_encoder = FakeTextEncoder(device=model.device_torch, dtype=model.torch_dtype)
 
     flush()