From fbac1cb7f50f86b539f5c56a7847ef8878df012f Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Sun, 24 May 2026 16:05:22 -0600 Subject: [PATCH] Dont force flash attention on hidream 01. Causes random issues and is slower. --- .../diffusion_models/hidream/hidream_o1_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/extensions_built_in/diffusion_models/hidream/hidream_o1_model.py b/extensions_built_in/diffusion_models/hidream/hidream_o1_model.py index 28f60e3d..41bf5e55 100644 --- a/extensions_built_in/diffusion_models/hidream/hidream_o1_model.py +++ b/extensions_built_in/diffusion_models/hidream/hidream_o1_model.py @@ -445,7 +445,7 @@ class HidreamO1Model(BaseModel): vinputs=z, timestep=t_pixeldit.reshape(-1), 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