RomanNum3ral
/
ai-toolkit
mirror of https://github.com/ostris/ai-toolkit.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' of github.com:ostris/ai-toolkit

This commit is contained in:
Jaret Burkett 2026-03-26 09:11:50 -06:00
parent 4df3b0463f 489b194231
commit 401f7df425
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
extensions_built_in/diffusion_models/flux2/flux2_model.py
View File

@ -412,8 +412,8 @@ class Flux2Model(BaseModel):
assert img_cond_seq_ids is not None, (
"You need to provide either both or neither of the sequence conditioning"
)
img_input = torch.cat((img_input, img_cond_seq), dim=1)
img_input_ids = torch.cat((img_input_ids, img_cond_seq_ids), dim=1)
img_input = torch.cat((img_input, img_cond_seq.to(img_input.device, img_input.dtype)), dim=1)
img_input_ids = torch.cat((img_input_ids, img_cond_seq_ids.to(img_input_ids.device)), dim=1)
guidance_vec = torch.full(
(img_input.shape[0],),