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

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

Browse Source
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
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"