JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2026-05-08 07:58:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge 9f548bce116de35a2e658d5e6a18558df1a99ea6 into 7a1af7154511e0ce4e4be8d62faa8c5e5a3532d2

Browse Source
This commit is contained in:
Harikrishna KP 2026-02-11 18:00:44 +05:30 committed by GitHub
commit 57bce4ac4b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
sat/sgm/modules/diffusionmodules/loss.py
View File

@ -121,7 +121,7 @@ class VideoDiffusionLoss(StandardDiffusionLoss):
w = append_dims(1 / (1 - alphas_cumprod_sqrt**2), input.ndim) # v-pred
if self.min_snr_value is not None:
w = min(w, self.min_snr_value)
w = torch.clamp(w, max=self.min_snr_value)
return self.get_loss(model_output, input, w)
def get_loss(self, model_output, target, w):