From 893b3f085e01dc1db224d9f983a7a82fb4f4d584 Mon Sep 17 00:00:00 2001 From: hehua2008 Date: Mon, 9 Dec 2024 15:29:48 +0800 Subject: [PATCH] Change Flux default max_shift to 1.15 to match the official one (#1137) --- flux/flux/sampler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flux/flux/sampler.py b/flux/flux/sampler.py index e7a1080d..6f293edc 100644 --- a/flux/flux/sampler.py +++ b/flux/flux/sampler.py @@ -7,7 +7,7 @@ import mlx.core as mx class FluxSampler: - def __init__(self, name: str, base_shift: float = 0.5, max_shift: float = 1.5): + def __init__(self, name: str, base_shift: float = 0.5, max_shift: float = 1.15): self._base_shift = base_shift self._max_shift = max_shift self._schnell = "schnell" in name