Merge pull request #98 from stanfordnlp/aryaman-runtime-fix

[P0] remove autograd debugging
stanfordnlp · Jan 30, 2024 · 2531a5f · 2531a5f
2 parents 8c0e544 + 9d747e8
commit 2531a5f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyvene/models/intervenable_base.py b/pyvene/models/intervenable_base.py
@@ -936,7 +936,7 @@ def _wait_for_forward_with_parallel_intervention(
         activations_sources: Optional[Dict] = None,
         subspaces: Optional[List] = None,
     ):
-        torch.autograd.set_detect_anomaly(True)
+        # torch.autograd.set_detect_anomaly(True)
         all_set_handlers = HandlerList([])
         unit_locations_sources = unit_locations["sources->base"][0]
         unit_locations_base = unit_locations["sources->base"][1]