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

Dont force flash attention on hidream 01. Causes random issues and is slower.

Browse Source
This commit is contained in:
Jaret Burkett
2026-05-24 16:05:22 -06:00
parent 307ff11bc5
commit fbac1cb7f5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extensions_built_in/diffusion_models/hidream/hidream_o1_model.py
View File

@@ -445,7 +445,7 @@ class HidreamO1Model(BaseModel):
vinputs=z, vinputs=z,
timestep=t_pixeldit.reshape(-1), timestep=t_pixeldit.reshape(-1),
token_types=token_types, token_types=token_types,
use_flash_attn=True, use_flash_attn=False,
) )
x_pred = outputs.x_pred # (B, S, C*p*p) over the full padded sequence x_pred = outputs.x_pred # (B, S, C*p*p) over the full padded sequence