diff --git a/pyvene/models/intervenable_base.py b/pyvene/models/intervenable_base.py index a946abe5..179bd600 100644 --- 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]