ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-30 11:11:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixe a variable that could nt be declared when doing blank prompt preservation

Browse Source
This commit is contained in:
Jaret Burkett
2025-10-23 16:19:56 -06:00
parent ee206cfa18
commit 80418209b8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extensions_built_in/sd_trainer/SDTrainer.py
View File

@@ -350,7 +350,7 @@ class SDTrainer(BaseSDTrainProcess):
if self.train_config.blank_prompt_preservation and self.cached_blank_embeds is None:
# make sure we have this if not unloading
self.cached_blank_embeds = self.sd.encode_prompt("", **encode_kwargs).to(
self.cached_blank_embeds = self.sd.encode_prompt("").to(
self.device_torch,
dtype=self.sd.torch_dtype
).detach()