From d14f49f758b99e371f45f3351e6c807487ef6cba Mon Sep 17 00:00:00 2001 From: Fabio Date: Sun, 18 Aug 2024 21:23:52 +0200 Subject: [PATCH] revert changes to config --- src/flux_1_schnell/config/config.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/flux_1_schnell/config/config.py b/src/flux_1_schnell/config/config.py index ca5a9f6..0335128 100644 --- a/src/flux_1_schnell/config/config.py +++ b/src/flux_1_schnell/config/config.py @@ -19,13 +19,14 @@ class Config: log.warning("Width and height should be multiples of 16. Rounding down.") self.width = 16 * (height // 16) self.height = 16 * (width // 16) - self.sigmas = Config.base_sigmas(num_inference_steps) + base_sigmas = Config.base_sigmas(num_inference_steps) self.num_inference_steps = num_inference_steps self.guidance = guidance + self.sigmas = mx.concatenate([base_sigmas, mx.zeros(1)]) @staticmethod def base_sigmas(num_inference_steps): - sigmas = np.linspace(1.0, 0, num_inference_steps+1) + sigmas = np.linspace(1.0, 1 / num_inference_steps, num_inference_steps) sigmas = mx.array(sigmas).astype(mx.float32) return sigmas