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

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

Browse Source
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
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, ( assert img_cond_seq_ids is not None, (
"You need to provide either both or neither of the sequence conditioning" "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 = 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), 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( guidance_vec = torch.full(
(img_input.shape[0],), (img_input.shape[0],),