From dc36bbb3c8c64073dc0f6adb7e1ebc89c921297e Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Mon, 23 Oct 2023 08:12:58 -0600 Subject: [PATCH] Added long prompts to general training --- extensions_built_in/sd_trainer/SDTrainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/extensions_built_in/sd_trainer/SDTrainer.py b/extensions_built_in/sd_trainer/SDTrainer.py index 3f7ecbc1..f594a7b4 100644 --- a/extensions_built_in/sd_trainer/SDTrainer.py +++ b/extensions_built_in/sd_trainer/SDTrainer.py @@ -285,7 +285,7 @@ class SDTrainer(BaseSDTrainProcess): with self.timer('encode_prompt'): if grad_on_text_encoder: 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, dtype=dtype) else: @@ -296,7 +296,7 @@ class SDTrainer(BaseSDTrainProcess): te.eval() else: 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, dtype=dtype)