ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-01-26 16:39:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix issue with fake text encoder unload

Browse Source
This commit is contained in:
Jaret Burkett
2025-08-14 09:33:44 -06:00
parent 5501521c9f
commit 3ff4430e84
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/unloader.py
View File

@@ -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()