From 50e5d995459832009c1ed49c810e57a457e82235 Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Wed, 19 Nov 2025 09:01:00 -0700
Subject: [PATCH] Fix issue where text encoder was not fully unloaded in some
 instances

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

diff --git a/toolkit/unloader.py b/toolkit/unloader.py
index 09ce1ba9..31f75e17 100644
--- a/toolkit/unloader.py
+++ b/toolkit/unloader.py
@@ -47,6 +47,7 @@ def unload_text_encoder(model: "BaseModel"):
             if hasattr(pipe, "text_encoder"):
                 te = FakeTextEncoder(device=model.device_torch, dtype=model.torch_dtype)
                 text_encoder_list.append(te)
+                pipe.text_encoder.to('cpu')
                 pipe.text_encoder = te
 
             i = 2