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

Initial support for finetuning qwen image. Will only work with caching for now, need to add controls everywhere.

Browse Source
This commit is contained in:
Jaret Burkett
2025-08-21 16:41:17 -06:00
parent 38d3814be7
commit bf2700f7be
12 changed files with 399 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
toolkit/data_loader.py
View File

@@ -497,6 +497,7 @@ class AiToolkitDataset(LatentCachingMixin, ControlCachingMixin, CLIPCachingMixin
dataloader_transforms=self.transform,
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,
)
self.file_list.append(file_item)
except Exception as e: