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

Remove timestep_index

Browse Source
This commit is contained in:
rockerBOO
2025-11-17 11:26:38 -05:00
parent 7a08c52aa4
commit cc0e4acf1b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
flux_train_network.py
View File

@@ -334,7 +334,7 @@ class FluxNetworkTrainer(train_network.NetworkTrainer):
# If CDC is enabled, this will transform the noise with geometry-aware covariance
noisy_model_input, timesteps, sigmas = flux_train_utils.get_noisy_model_input_and_timestep(
args, noise_scheduler, latents, noise, accelerator.device, weight_dtype,
gamma_b_dataset=gamma_b_dataset, latents_npz_paths=latents_npz_paths, timestep_index=timestep_index
gamma_b_dataset=gamma_b_dataset, latents_npz_paths=latents_npz_paths
)
# pack latents and get img_ids