github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 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

@@ -176,10 +176,6 @@ class BaseSDTrainProcess(BaseTrainProcess):
dataset.cache_text_embeddings for dataset in self.dataset_configs 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 self.embed_config = None
embedding_raw = self.get_conf('embedding', None) embedding_raw = self.get_conf('embedding', None)
if embedding_raw is not None: if embedding_raw is not None:

2
version.py
View File

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