Merge pull request #186 from stanfordnlp/zen/quick_fix

[Minor] Fix intervention return objects
stanfordnlp · Sep 6, 2024 · 76bb067 · 76bb067
2 parents d4ca094 + 95b74ee
commit 76bb067
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyvene/models/interventions.py b/pyvene/models/interventions.py
@@ -18,7 +18,7 @@ class InterventionOutput(ModelOutput):
     Output of the IntervenableModel, including original outputs, intervened outputs, and collected activations.
     """
     output: Optional[Any] = None
-    loss: Optional[Any] = None
+    latent: Optional[Any] = None
 
 
 class Intervention(torch.nn.Module):