[Minor] Update use_cache intervenable_base.py

stanfordnlp · Jul 22, 2024 · 312c38f · 312c38f
1 parent 0e11417
commit 312c38f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyvene/models/intervenable_base.py b/pyvene/models/intervenable_base.py
@@ -1781,7 +1781,7 @@ def forward(
         labels: Optional[torch.LongTensor] = None,
         output_original_output: Optional[bool] = False,
         return_dict: Optional[bool] = None,
-        use_cache: Optional[bool] = True,
+        use_cache: Optional[bool] = None,
     ):
         """
         Main forward function that serves a wrapper to