From a12ddd72a1a0a07ef868f7be6cfccc769e97f200 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Thu, 7 May 2026 07:37:05 -0600 Subject: [PATCH] Change the velocity weight cap on dfe 9 --- toolkit/models/diffusion_feature_extraction.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/toolkit/models/diffusion_feature_extraction.py b/toolkit/models/diffusion_feature_extraction.py index 4d8489ba..7332a123 100644 --- a/toolkit/models/diffusion_feature_extraction.py +++ b/toolkit/models/diffusion_feature_extraction.py @@ -1150,7 +1150,7 @@ class DiffusionFeatureExtractor9(nn.Module): perceptual_loss = torch.nn.functional.mse_loss( pred.float(), target.float(), reduction="none" ) - velocity_equiv_weight = (1.0 / torch.clamp(tv, min=0.001) ** 2) + velocity_equiv_weight = (1.0 / torch.clamp(tv, min=0.1) ** 2) loss_perceptual = (perceptual_loss * velocity_equiv_weight).mean() if self.do_partial_step: