Update reference.py

iamgroot42 · Feb 7, 2024 · 4a56bf8 · 4a56bf8
1 parent 9f61a49
commit 4a56bf8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mimir/attacks/reference.py b/mimir/attacks/reference.py
@@ -19,4 +19,4 @@ def _attack(self, document, probs, tokens=None, **kwargs):
         if loss is None:
             loss = self.model.get_ll(document, probs=probs, tokens=tokens)
         ref_loss = self.ref_model.get_ll(document, probs=probs, tokens=tokens)
-        return ref_loss - loss
+        return loss - ref_loss