DarthAffe/stable-diffusion.cpp
1
0 Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-11-11 11:26:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: use patch token count for the Flux scheduler (#2293)

Browse Source
This commit is contained in:
DrMkdaddy authored and GitHub committed 2026-11-08 21:37:06 +08:00
1 parent f61053c25d
commit b43af40461
3 files changed
+14 -1
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/pipeline/diffusion_engine.cpp
+6 -1
View File
@@ -2930,7 +2930,12 @@ int StableDiffusionGGML::get_image_channels() const {
int StableDiffusionGGML::get_image_seq_len(int h, int w) {
int vae_scale_factor = get_vae_scale_factor();
return (h / vae_scale_factor) * (w / vae_scale_factor);
int patch_size = diffusion_model != nullptr ? diffusion_model->get_patch_size() : 1;
int latent_h = h / vae_scale_factor;
int latent_w = w / vae_scale_factor;
int tokens_h = (latent_h + patch_size - 1) / patch_size;
int tokens_w = (latent_w + patch_size - 1) / patch_size;
return tokens_h * tokens_w;
}
sd::Tensor<float> StableDiffusionGGML::generate_init_latent(int width,