VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-15 08:36:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update flux_train_utils.py

Browse Source
This commit is contained in:
sdbds
2024-09-05 14:57:55 +08:00
parent 9af36fabe7
commit 2f2473c432
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
library/flux_train_utils.py
View File

@@ -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=x1 / 256 * get_lin_function(x1),
y2=4096 / x2 * 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)