ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-05-01 03:31:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixes to allow for redux assisted training

Browse Source
This commit is contained in:
Jaret Burkett
2025-03-03 16:27:19 -07:00
parent 1f3f45a48d
commit c5e0c2bbe2
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
toolkit/stable_diffusion_model.py
View File

@@ -1343,7 +1343,8 @@ class StableDiffusion:
conditional_clip_embeds = self.adapter.get_clip_image_embeds_from_tensors(validation_image) conditional_clip_embeds = self.adapter.get_clip_image_embeds_from_tensors(validation_image)
self.adapter(conditional_clip_embeds) 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 # handle condition the prompts
gen_config.prompt = self.adapter.condition_prompt( gen_config.prompt = self.adapter.condition_prompt(
gen_config.prompt, gen_config.prompt,
@@ -1397,7 +1398,7 @@ class StableDiffusion:
conditional_embeds = self.adapter(conditional_embeds, conditional_clip_embeds, is_unconditional=False) conditional_embeds = self.adapter(conditional_embeds, conditional_clip_embeds, is_unconditional=False)
unconditional_embeds = self.adapter(unconditional_embeds, unconditional_clip_embeds, is_unconditional=True) 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( conditional_embeds = self.adapter.condition_encoded_embeds(
tensors_0_1=validation_image, tensors_0_1=validation_image,
prompt_embeds=conditional_embeds, prompt_embeds=conditional_embeds,