From d932b97597f862e102f7a0235fe879b8ee731a99 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=9D=92=E9=BE=8D=E8=81=96=E8=80=85=40bdsqlsz?= Date: Thu, 5 Sep 2024 16:38:08 +0800 Subject: [PATCH] fix function --- library/flux_train_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/library/flux_train_utils.py b/library/flux_train_utils.py index 892b4851..51ef1765 100644 --- a/library/flux_train_utils.py +++ b/library/flux_train_utils.py @@ -414,8 +414,8 @@ def get_noisy_model_input_and_timesteps( timesteps = logits_norm.sigmoid() x1, x2 = args.min_bucket_reso or min(args.resolution), args.max_bucket_reso or max(args.resolution) mu = get_lin_function( - y1=get_lin_function(x1), - y2=get_lin_function(x2), + y1=get_lin_function()(x1), + y2=get_lin_function()(x2), )((h // 2) * (w // 2)) timesteps = time_shift(mu, 1.0, timesteps)