ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-03-12 05:59:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

always prompt both even when training only one encoder

Browse Source
This commit is contained in:
Jaret Burkett
2023-10-07 11:23:09 -06:00
parent cac8754399
commit b1cfafa0c6
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
toolkit/stable_diffusion_model.py
View File

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