Fix issue with fake text encoder unload

2025-08-14 09:33:44 -06:00 · 2025-08-14 09:33:44 -06:00 · 3ff4430e84
parent 5501521c9f
commit 3ff4430e84
1 changed files with 1 additions and 1 deletions
--- 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()