From db581cea2201ce7fa7c5f2ce22c10e485292402d Mon Sep 17 00:00:00 2001 From: Ajay Gopi Date: Wed, 16 Oct 2024 12:47:46 -0400 Subject: [PATCH] fix module usage guide --- pyvene_101.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pyvene_101.ipynb b/pyvene_101.ipynb index 07bc46d0..5da0e240 100644 --- a/pyvene_101.ipynb +++ b/pyvene_101.ipynb @@ -2598,7 +2598,7 @@ " model=resnet\n", ")\n", "intervened_outputs = pv_resnet(\n", - " base_inputs, [source_inputs], return_dict=True\n", + " base_inputs, [source_inputs], output_original_output = True, return_dict=True\n", ")\n", "(intervened_outputs.intervened_outputs.logits - intervened_outputs.original_outputs.logits).sum()" ] @@ -2655,7 +2655,7 @@ ")\n", "\n", "intervened_outputs = pv_resnet(\n", - " base_inputs, [source_inputs], return_dict=True\n", + " base_inputs, [source_inputs], output_original_output = True, return_dict=True\n", ")\n", "(intervened_outputs.intervened_outputs.logits - intervened_outputs.original_outputs.logits).sum()" ]