ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-05-11 16:30:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change the velocity weight cap on dfe 9

Browse Source
This commit is contained in:
Jaret Burkett
2026-05-07 07:37:05 -06:00
parent 6bb8acbffc
commit a12ddd72a1
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/models/diffusion_feature_extraction.py
View File

@@ -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: