From 73ebae13a468942ded8b47991b5d56eeeea25c36 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Sat, 24 Oct 2026 10:12:38 -0600 Subject: [PATCH] Increase the scale range of DFEs velocity scale correction --- toolkit/models/diffusion_feature_extraction.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/toolkit/models/diffusion_feature_extraction.py b/toolkit/models/diffusion_feature_extraction.py index 910bfbc..45c2570 100644 --- a/toolkit/models/diffusion_feature_extraction.py +++ b/toolkit/models/diffusion_feature_extraction.py @@ -1221,7 +1221,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.1) ** 2) + velocity_equiv_weight = (1.0 / torch.clamp(tv, min=0.01) ** 2) loss_perceptual = (perceptual_loss * velocity_equiv_weight).mean() if self.do_partial_step: @@ -1379,7 +1379,7 @@ class DiffusionFeatureExtractor10(nn.Module): pred_feats = self.get_lpips_features(pred_images.float()) - velocity_equiv_weight = (1.0 / torch.clamp(tv, min=0.1) ** 2) + velocity_equiv_weight = (1.0 / torch.clamp(tv, min=0.01) ** 2) loss_perceptual = 0 for idx, pred_feat in enumerate(pred_feats):