Skip to content

Commit 0d9d5e0

Browse files
committed
Merge branch 'master' into chroma_radiance
2 parents 458c365 + 69b9511 commit 0d9d5e0

File tree

2 files changed

+9
-6
lines changed

2 files changed

+9
-6
lines changed

ggml

Submodule ggml updated 59 files

stable-diffusion.cpp

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -765,11 +765,14 @@ class StableDiffusionGGML {
765765
denoiser = std::make_shared<DiscreteFlowDenoiser>(shift);
766766
} else if (sd_version_is_flux(version)) {
767767
LOG_INFO("running in Flux FLOW mode");
768-
float shift = 1.0f; // TODO: validate
769-
for (auto pair : model_loader.tensor_storages_types) {
770-
if (pair.first.find("model.diffusion_model.guidance_in.in_layer.weight") != std::string::npos) {
771-
shift = 1.15f;
772-
break;
768+
float shift = sd_ctx_params->flow_shift;
769+
if (shift == INFINITY) {
770+
shift = 1.0f; // TODO: validate
771+
for (auto pair : model_loader.tensor_storages_types) {
772+
if (pair.first.find("model.diffusion_model.guidance_in.in_layer.weight") != std::string::npos) {
773+
shift = 1.15f;
774+
break;
775+
}
773776
}
774777
}
775778
denoiser = std::make_shared<FluxFlowDenoiser>(shift);

0 commit comments

Comments
 (0)