ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-01-26 16:39:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Integrate dataset level trigger words and allow them to be cached. Default to global trigger if it is set.

Browse Source
This commit is contained in:
Jaret Burkett
2025-09-18 03:29:18 -06:00
parent 3cdf50cbfc
commit 390e21bec6
2 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
jobs/process/BaseSDTrainProcess.py
View File

@@ -156,6 +156,9 @@ class BaseSDTrainProcess(BaseTrainProcess):
if raw_datasets is not None and len(raw_datasets) > 0:
for raw_dataset in raw_datasets:
dataset = DatasetConfig(**raw_dataset)
# handle trigger word per dataset
if dataset.trigger_word is None and self.trigger_word is not None:
dataset.trigger_word = self.trigger_word
is_caching = dataset.cache_latents or dataset.cache_latents_to_disk
if not is_caching:
self.is_latents_cached = False