diff --git a/pyvene/models/intervenable_base.py b/pyvene/models/intervenable_base.py index 60a476f7..db84b57d 100644 --- a/pyvene/models/intervenable_base.py +++ b/pyvene/models/intervenable_base.py @@ -1570,7 +1570,6 @@ def hook_callback(model, args, kwargs, output=None): subspaces[key_i] if subspaces is not None else None, ) if isinstance(raw_intervened_representation, InterventionOutput): - # memorize for other training objectives self.full_intervention_outputs.append(raw_intervened_representation) intervened_representation = raw_intervened_representation.output else: