feat: add sd3 flash attn support (#815)

2026-07-30 06:40:41 -05:00 · 2025-09-11 23:24:29 +08:00
parent 49d6570c43
commit fce6afcc6a
3 changed files with 36 additions and 21 deletions
--- a/stable-diffusion.cpp
+++ b/stable-diffusion.cpp
@@ -350,6 +350,7 @@ public:
                                                                     model_loader.tensor_storages_types);
                diffusion_model  = std::make_shared<MMDiTModel>(backend,
                                                               offload_params_to_cpu,
+                                                               sd_ctx_params->diffusion_flash_attn,
                                                               model_loader.tensor_storages_types);
            } else if (sd_version_is_flux(version)) {
                bool is_chroma = false;