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 train_network.py

Browse Source
This commit is contained in:
DKnight54
2025-03-28 19:29:59 +08:00
committed by GitHub
parent 666468857b
commit 4456065d29
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train_network.py
View File

@@ -929,7 +929,7 @@ class NetworkTrainer:
if torch.any(torch.isnan(latents)):
accelerator.print("NaN found in latents, replacing with zeros")
latents = torch.nan_to_num(latents, 0, out=latents)
logger.info(f"Latents Shape: {latents.shape}")
logger.info(f"Latents Shape: {latents.shape}")
latents = latents * self.vae_scale_factor
# get multiplier for each sample