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

Update train_network.py

Browse Source
This commit is contained in:
DKnight54
2025-02-03 05:22:00 +08:00
committed by GitHub
parent e250e7aad0
commit 4c5ec694f7
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
train_network.py
View File

@@ -1024,9 +1024,9 @@ class NetworkTrainer:
# Checks if the accelerator has performed an optimization step behind the scenes
# Collecting latents and caption lists from all processes
logger.info(f"latents.size: {latents.size()} before gather on device {accelerator.state.local_process_index}")
#Converts batch of latents into list of dicts containing individual latents, height and width to merge across processes
#Allows for different latent sizes
# Converts batch of latents into list of dicts containing individual latents, height and width to merge across processes
# Allows for different latent sizes
latents_list = []
for idx in range(len(batch["captions"])):
latent_dict = {}