Compare commits

...

2 Commits

Author SHA1 Message Date
leejet
69b9511ce9 sync: update ggml 2025-10-24 00:32:45 +08:00
stduhpf
917f7bfe99
fix: support --flow-shift for flux models with default pred (#913) 2025-10-23 21:35:18 +08:00
2 changed files with 9 additions and 6 deletions

2
ggml

@ -1 +1 @@
Subproject commit c538174d261d8172480f87efcfec8e69aac13ebb
Subproject commit 2d3876d554551d35c06dccc5852be50d5fd2a275

View File

@ -761,11 +761,14 @@ public:
denoiser = std::make_shared<DiscreteFlowDenoiser>(shift);
} else if (sd_version_is_flux(version)) {
LOG_INFO("running in Flux FLOW mode");
float shift = 1.0f; // TODO: validate
for (auto pair : model_loader.tensor_storages_types) {
if (pair.first.find("model.diffusion_model.guidance_in.in_layer.weight") != std::string::npos) {
shift = 1.15f;
break;
float shift = sd_ctx_params->flow_shift;
if (shift == INFINITY) {
shift = 1.0f; // TODO: validate
for (auto pair : model_loader.tensor_storages_types) {
if (pair.first.find("model.diffusion_model.guidance_in.in_layer.weight") != std::string::npos) {
shift = 1.15f;
break;
}
}
}
denoiser = std::make_shared<FluxFlowDenoiser>(shift);