RomanNum3ral
/
ai-toolkit
mirror of https://github.com/ostris/ai-toolkit.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Allow trigger when caching text embeddings since it is now passed to dataset

This commit is contained in:
Jaret Burkett 2025-09-30 16:58:35 -06:00
parent 2ba4000704
commit b07b88c46b
2 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
jobs/process/BaseSDTrainProcess.py
View File

@ -175,10 +175,6 @@ class BaseSDTrainProcess(BaseTrainProcess):
self.is_caching_text_embeddings = any(
dataset.cache_text_embeddings for dataset in self.dataset_configs
)
# cannot train trigger word if caching text embeddings
if self.is_caching_text_embeddings and self.trigger_word is not None:
raise ValueError("Cannot train trigger word if caching text embeddings. Please remove the trigger word or disable text embedding caching.")
self.embed_config = None
embedding_raw = self.get_conf('embedding', None)

2
version.py
View File

@ -1 +1 @@
VERSION = "0.6.0"
VERSION = "0.6.1"