Allow using prior loss with a custom adapter

2025-03-01 08:01:14 -07:00 · 2025-03-01 08:01:14 -07:00 · 60539c0b0f
parent dd700f70b3
commit 60539c0b0f
1 changed files with 1 additions and 1 deletions
--- a/extensions_built_in/sd_trainer/SDTrainer.py
+++ b/extensions_built_in/sd_trainer/SDTrainer.py
@ -895,7 +895,7 @@ class SDTrainer(BaseSDTrainProcess):
            was_adapter_active = self.adapter.is_active
            self.adapter.is_active = False

-        if self.train_config.unload_text_encoder and self.adapter is not None:
+        if self.train_config.unload_text_encoder and self.adapter is not None and not isinstance(self.adapter, CustomAdapter):
            raise ValueError("Prior predictions currently do not support unloading text encoder with adapter")
        # do a prediction here so we can match its output with network multiplier set to 0.0
        with torch.no_grad():