From c5e0c2bbe22889962f72005514234b6cd63df531 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Mon, 3 Mar 2025 16:27:19 -0700 Subject: [PATCH] Fixes to allow for redux assisted training --- toolkit/stable_diffusion_model.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/toolkit/stable_diffusion_model.py b/toolkit/stable_diffusion_model.py index 00b38574..7a2dcdff 100644 --- a/toolkit/stable_diffusion_model.py +++ b/toolkit/stable_diffusion_model.py @@ -1343,7 +1343,8 @@ class StableDiffusion: conditional_clip_embeds = self.adapter.get_clip_image_embeds_from_tensors(validation_image) self.adapter(conditional_clip_embeds) - if self.adapter is not None and isinstance(self.adapter, CustomAdapter): + if self.adapter is not None and isinstance(self.adapter, CustomAdapter) \ + and gen_config.adapter_image_path is not None: # handle condition the prompts gen_config.prompt = self.adapter.condition_prompt( gen_config.prompt, @@ -1397,7 +1398,7 @@ class StableDiffusion: conditional_embeds = self.adapter(conditional_embeds, conditional_clip_embeds, is_unconditional=False) unconditional_embeds = self.adapter(unconditional_embeds, unconditional_clip_embeds, is_unconditional=True) - if self.adapter is not None and isinstance(self.adapter, CustomAdapter): + if self.adapter is not None and isinstance(self.adapter, CustomAdapter) and validation_image is not None: conditional_embeds = self.adapter.condition_encoded_embeds( tensors_0_1=validation_image, prompt_embeds=conditional_embeds,