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

Added residual

Browse Source
This commit is contained in:
CrossProduct
2025-01-22 20:10:01 +00:00
parent 981767154b
commit 731f326e86
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
finetrainers/trainer.py
+1 -1
View File
@@ -827,6 +827,7 @@ class Trainer:
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"]})
else:
# Default to flow-matching noise addition
noisy_latents = (1.0 - sigmas) * latent_conditions["latents"] + sigmas * noise
@@ -848,7 +849,6 @@ class Trainer:
transformer=self.transformer,
scheduler=self.scheduler,
timesteps=timesteps,
residual_x=noisy_latents_residual,
**latent_conditions,
**text_conditions,
)