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

Merge pull request #25 from rockerBOO/lumina-fix-non-cache-image-vae-encode

Browse Source 
Fix non-cache vae encode
This commit is contained in:
青龍聖者@bdsqlsz
2025-03-22 14:58:52 +08:00
committed by GitHub
parent 7482784f74 61f7283167
commit 1481217eb2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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