diff --git a/jobs/process/BaseSDTrainProcess.py b/jobs/process/BaseSDTrainProcess.py index 7e8d0576..e63f899c 100644 --- a/jobs/process/BaseSDTrainProcess.py +++ b/jobs/process/BaseSDTrainProcess.py @@ -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) diff --git a/version.py b/version.py index af9f7a48..01b25ca2 100644 --- a/version.py +++ b/version.py @@ -1 +1 @@ -VERSION = "0.6.0" \ No newline at end of file +VERSION = "0.6.1" \ No newline at end of file