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

Merge 79ed5177eb into 63711390a0

Browse Source
This commit is contained in:
Elthariel
2025-09-30 09:55:44 +05:30
committed by GitHub
parent 63711390a0 79ed5177eb
commit 9a02c4658e
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
train_controlnet.py
View File

@@ -446,8 +446,11 @@ def train(args):
)
# Sample a random timestep for each image
min_timestep = args.min_timestep if args.min_timestep is not None else 0
sched_max_timestep = noise_scheduler.config.num_train_timesteps
max_timestep = args.max_timestep if args.max_timestep is not None else sched_max_timestep
timesteps, huber_c = train_util.get_timesteps_and_huber_c(
args, 0, noise_scheduler.config.num_train_timesteps, noise_scheduler, b_size, latents.device
args, min_timestep, max_timestep, noise_scheduler, b_size, latents.device
)
# Add noise to the latents according to the noise magnitude at each timestep