Allow dataloader to encode first frame with the text embeddings if the model needs it.

This commit is contained in:
Jaret Burkett committed 2026-07-29 17:50:45 -06:00
1 parent 65443cfffa
commit 3f8afcac7e
3 files changed
+32

No files matched your search

+1
View File
@@ -533,6 +533,7 @@ class AiToolkitDataset(LatentCachingMixin, ControlCachingMixin, CLIPCachingMixin
size_database=self.size_database,
dataset_root=dataset_folder,
encode_control_in_text_embeddings=self.sd.encode_control_in_text_embeddings if self.sd else False,
encode_first_frame_in_text_embeddings=getattr(self.sd, 'encode_first_frame_in_text_embeddings', False) if self.sd else False,
text_embedding_space_version=self.sd.text_embedding_space_version if self.sd else "sd1",
te_padding_side=self.sd.te_padding_side if self.sd else "right",
latent_space_version=latent_space_version,