DarthAffe/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-01-02 10:43:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: make flash attn work with high noise diffusion model (#1111)

Browse Source
This commit is contained in:
leejet 2025-12-17 23:28:59 +08:00 committed by GitHub
parent c3ad6a13e1
commit c2e18c86e8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
stable-diffusion.cpp
View File

@ -537,6 +537,9 @@ public:
if (sd_ctx_params->diffusion_flash_attn) {
LOG_INFO("Using flash attention in the diffusion model");
diffusion_model->set_flash_attn_enabled(true);
if (high_noise_diffusion_model) {
high_noise_diffusion_model->set_flash_attn_enabled(true);
}
}
cond_stage_model->alloc_params_buffer();
@ -3723,6 +3726,9 @@ SD_API sd_image_t* generate_video(sd_ctx_t* sd_ctx, const sd_vid_gen_params_t* s
init_latent = sd_ctx->sd->generate_init_latent(work_ctx, width, height, frames, true);
}
print_ggml_tensor(init_latent, true);
print_ggml_tensor(concat_latent, true);
// Get learned condition
ConditionerParams condition_params;
condition_params.clip_skip = sd_vid_gen_params->clip_skip;