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

Merge branch 'lumina' of https://github.com/sdbds/sd-scripts into lumina

Browse Source
This commit is contained in:
sdbds
2025-04-06 16:09:37 +08:00
parent 00e12eed65 30008168e3
commit 1a4f1ff0f1
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
library/lumina_train_util.py
View File

@@ -688,9 +688,9 @@ def denoise(
noise_pred, dim=tuple(range(1, len(noise_pred.shape))), keepdim=True
)
# Iterate through batch
for noise_norm, max_new_norm, noise in zip(noise_norms, max_new_norms, noise_pred):
for i, (noise_norm, max_new_norm) in enumerate(zip(noise_norms, max_new_norms)):
if noise_norm >= max_new_norm:
noise = noise * (max_new_norm / noise_norm)
noise_pred[i] = noise_pred[i] * (max_new_norm / noise_norm)
else:
noise_pred = noise_pred_cond

2
lumina_train_network.py
View File

@@ -230,7 +230,7 @@ class LuminaNetworkTrainer(train_network.NetworkTrainer):
self.noise_scheduler_copy = copy.deepcopy(noise_scheduler)
return noise_scheduler
def encode_images_to_latents(self, args, accelerator, vae, images):
def encode_images_to_latents(self, args, vae, images):
return vae.encode(images)
# not sure, they use same flux vae