github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Added long prompts to general training

Browse Source
This commit is contained in:
Jaret Burkett
2023-10-23 08:12:58 -06:00
parent 9905a1e205
commit dc36bbb3c8
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
extensions_built_in/sd_trainer/SDTrainer.py
View File

@@ -285,7 +285,7 @@ class SDTrainer(BaseSDTrainProcess):
with self.timer('encode_prompt'): with self.timer('encode_prompt'):
if grad_on_text_encoder: if grad_on_text_encoder:
with torch.set_grad_enabled(True): with torch.set_grad_enabled(True):
conditional_embeds = self.sd.encode_prompt(conditioned_prompts).to( conditional_embeds = self.sd.encode_prompt(conditioned_prompts, long_prompts=True).to(
self.device_torch, self.device_torch,
dtype=dtype) dtype=dtype)
else: else:
@@ -296,7 +296,7 @@ class SDTrainer(BaseSDTrainProcess):
te.eval() te.eval()
else: else:
self.sd.text_encoder.eval() self.sd.text_encoder.eval()
conditional_embeds = self.sd.encode_prompt(conditioned_prompts).to( conditional_embeds = self.sd.encode_prompt(conditioned_prompts, long_prompts=True).to(
self.device_torch, self.device_torch,
dtype=dtype) dtype=dtype)