Skip to content

Commit 7ae38dd

Browse files
Block sourceless training with text encoder training
Text encoder training requires re-tokenizing prompts from source files, which are not available in sourceless mode. Raise a clear error at dataset creation time rather than failing mid-training.
1 parent da0bead commit 7ae38dd

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

modules/dataLoader/mixin/DataLoaderText2ImageMixin.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -398,6 +398,11 @@ def _create_dataset(
398398
output_modules = self._output_modules(config, model, model_setup)
399399

400400
if config.sourceless_training and config.latent_caching:
401+
if hasattr(config, 'train_text_encoder_or_embedding') and config.train_text_encoder_or_embedding():
402+
raise RuntimeError(
403+
"Sourceless training cannot be used with text encoder training. "
404+
"Disable sourceless_training or disable text encoder training."
405+
)
401406
return self._create_mgds(
402407
config,
403408
[cache_modules, output_modules],

0 commit comments

Comments
 (0)