github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Improved the prompt handeling of ideogram4 model. Now used advanced prompts class to store them smaller and allow longer prompts

Browse Source
This commit is contained in:
Jaret Burkett
2026-06-04 13:00:23 -06:00
parent 2ddc2e1318
commit 6a8d9333b6
5 changed files with 75 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/data_loader.py
View File

@@ -533,7 +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,
text_embedding_space_version=self.sd.model_config.arch if self.sd else "sd1",
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,
temporal_compression=temporal_compression,