From 60539c0b0fe0b9ae93773ea309c91b5974ac5150 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Sat, 1 Mar 2025 08:01:14 -0700 Subject: [PATCH] Allow using prior loss with a custom adapter --- extensions_built_in/sd_trainer/SDTrainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/extensions_built_in/sd_trainer/SDTrainer.py b/extensions_built_in/sd_trainer/SDTrainer.py index 32f0c2fb..827e05c4 100644 --- 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():