From b1cfafa0c6ec787cc3cc51218346c275c7f1125a Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Sat, 7 Oct 2023 11:23:09 -0600 Subject: [PATCH] always prompt both even when training only one encoder --- toolkit/stable_diffusion_model.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/toolkit/stable_diffusion_model.py b/toolkit/stable_diffusion_model.py index 8d3ae4c4..fd2e6dfd 100644 --- a/toolkit/stable_diffusion_model.py +++ b/toolkit/stable_diffusion_model.py @@ -681,10 +681,10 @@ class StableDiffusion: # todo make this a config # 50% chance to use an encoder anyway even if it is disabled # allows the other TE to compensate for the disabled one - use_encoder_1 = self.use_text_encoder_1 or force_all or random.random() > 0.5 - use_encoder_2 = self.use_text_encoder_2 or force_all or random.random() > 0.5 - # use_encoder_1 = True - # use_encoder_2 = True + # use_encoder_1 = self.use_text_encoder_1 or force_all or random.random() > 0.5 + # use_encoder_2 = self.use_text_encoder_2 or force_all or random.random() > 0.5 + use_encoder_1 = True + use_encoder_2 = True return PromptEmbeds( train_tools.encode_prompts_xl(