manbeast3b
/

supasubmitsoon

Model card Files Files and versions

manbeast3b commited on Nov 24, 2024

Commit

697f86b

·

verified ·

1 Parent(s): e8fcb7e

Update src/pipeline.py

Files changed (1) hide show

src/pipeline.py +1 -1

src/pipeline.py CHANGED Viewed

@@ -122,7 +122,7 @@ class LossSchedulerModel(torch.nn.Module):
             mul = D*K
             norm_DE = torch.norm(mul)
             C+= (mul) / (norm_DE + 1e-8)
-            C+= E
         return C.to(xT.dtype)

             mul = D*K
             norm_DE = torch.norm(mul)
             C+= (mul) / (norm_DE + 1e-8)
+            C+= D*E
         return C.to(xT.dtype)