storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-09 00:09:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Post patchify the latents.

Browse Source
This commit is contained in:
CrossProduct
2025-01-22 21:24:33 +00:00
parent 731f326e86
commit d631737329
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
finetrainers/trainer.py
+9
View File
@@ -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"]})