diff --git a/extensions_built_in/sd_trainer/SDTrainer.py b/extensions_built_in/sd_trainer/SDTrainer.py index 178534a..68aa8ca 100644 --- a/extensions_built_in/sd_trainer/SDTrainer.py +++ b/extensions_built_in/sd_trainer/SDTrainer.py @@ -350,7 +350,7 @@ class SDTrainer(BaseSDTrainProcess): if self.train_config.blank_prompt_preservation and self.cached_blank_embeds is None: # make sure we have this if not unloading - self.cached_blank_embeds = self.sd.encode_prompt("", **encode_kwargs).to( + self.cached_blank_embeds = self.sd.encode_prompt("").to( self.device_torch, dtype=self.sd.torch_dtype ).detach()