github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Add support for training Z-Image Turbo with a de-distill training adapter

Browse Source
This commit is contained in:
Jaret Burkett
2025-11-28 08:08:53 -07:00
parent 21bb8a2bf4
commit 4e62c38df5
11 changed files with 459 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
toolkit/prompt_utils.py
View File

@@ -72,7 +72,10 @@ class PromptEmbeds:
if self.pooled_embeds is not None:
prompt_embeds = PromptEmbeds([cloned_text_embeds, self.pooled_embeds.clone()])
else:
prompt_embeds = PromptEmbeds(cloned_text_embeds)
if isinstance(cloned_text_embeds, list) or isinstance(cloned_text_embeds, tuple):
prompt_embeds = PromptEmbeds([cloned_text_embeds, None])
else:
prompt_embeds = PromptEmbeds(cloned_text_embeds)
if self.attention_mask is not None:
if isinstance(self.attention_mask, list) or isinstance(self.attention_mask, tuple):