From d631737329096ccc2dace24f332af564e55c52fb Mon Sep 17 00:00:00 2001 From: CrossProduct Date: Wed, 22 Jan 2025 21:24:33 +0000 Subject: [PATCH] Post patchify the latents. --- finetrainers/trainer.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/finetrainers/trainer.py b/finetrainers/trainer.py index 0e87537..3e2c259 100644 --- a/finetrainers/trainer.py +++ b/finetrainers/trainer.py @@ -825,6 +825,15 @@ class Trainer: width=latent_conditions["width"], patch_size = 1, patch_size_t = 1) + # target + latent_conditions["latents"] = post_conditioned_latent_patchify( + latents=latent_conditions["latents"], + num_frames=latent_conditions["num_frames"], + height=latent_conditions["height"], + width=latent_conditions["width"], + patch_size = 1, + patch_size_t = 1 + ) # pose information latent latent_conditions.update({"noisy_latents": latent_final["latents"]}) latent_conditions.update({"noisy_latents_residual":noisy_latents_residual["latents"]})