ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-26 09:19:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added initial support for f-lite model

Browse Source
This commit is contained in:
Jaret Burkett
2025-05-01 11:15:18 -06:00
parent 5890e67a46
commit d9700bdb99
9 changed files with 1076 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/train_tools.py
View File

@@ -145,7 +145,7 @@ if TYPE_CHECKING:
def concat_prompt_embeddings(
unconditional: 'PromptEmbeds',
conditional: 'PromptEmbeds',
n_imgs: int,
n_imgs: int=0,
):
from toolkit.stable_diffusion_model import PromptEmbeds
text_embeds = torch.cat(