Fix non-cache vae encode

This commit is contained in:
rockerBOO
2025-03-21 20:38:43 -04:00
parent 7482784f74
commit 61f7283167

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