VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix errors in SD3 LoRA training with Text Encoders close #1724

Browse Source
This commit is contained in:
kohya-ss
2024-10-26 17:29:24 +09:00
parent 0031d916f0
commit 56bf761164
2 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
sd3_train_network.py
View File

@@ -134,7 +134,7 @@ class Sd3NetworkTrainer(train_network.NetworkTrainer):
def get_models_for_text_encoding(self, args, accelerator, text_encoders):
if args.cache_text_encoder_outputs:
if self.train_clip and not self.train_t5xxl:
return text_encoders[0:2] # only CLIP-L/CLIP-G is needed for encoding because T5XXL is cached
return text_encoders[0:2] + [None] # only CLIP-L/CLIP-G is needed for encoding because T5XXL is cached
else:
return None # no text encoders are needed for encoding because both are cached
else: