Merge pull request #60 from stanfordnlp/zen/causaltracingfull

[Minor] Add in full causal tracing fig 1 reproduced results
stanfordnlp · Jan 17, 2024 · b223199 · b223199
2 parents 5894b18 + bfad1a4
commit b223199
Show file tree

Hide file tree

Showing 2 changed files with 208 additions and 108 deletions.
diff --git a/pyvene/__init__.py b/pyvene/__init__.py
@@ -20,6 +20,10 @@
 from .models.interventions import LowRankRotatedSpaceIntervention
 from .models.interventions import PCARotatedSpaceIntervention
 from .models.interventions import CollectIntervention
+from .models.interventions import ConstantSourceIntervention
+from .models.interventions import ZeroIntervention
+from .models.interventions import LocalistRepresentationIntervention
+from .models.interventions import DistributedRepresentationIntervention
 
 
 # Utils

diff --git a/tutorials/advanced_tutorials/Causal_Tracing.ipynb b/tutorials/advanced_tutorials/Causal_Tracing.ipynb