From 3fb6b710f2b8f3bfd6a54a3218c8e06e1830921e Mon Sep 17 00:00:00 2001 From: Arthur Date: Mon, 30 Jun 2025 16:49:11 +0200 Subject: [PATCH] update --- src/transformers/utils/generic.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/transformers/utils/generic.py b/src/transformers/utils/generic.py index 4386bae389e..ea5d4f82b95 100644 --- a/src/transformers/utils/generic.py +++ b/src/transformers/utils/generic.py @@ -1015,7 +1015,8 @@ def check_model_inputs(func): return capture_fn capture_flags = self._can_record_outputs - all_args.update(**all_args["kwargs"]) + if "kwargs" in all_args: + all_args.update(**all_args["kwargs"]) recordable_keys = { f"output_{k}": all_args.get(f"output_{k}", getattr(self.config, f"output_{k}", False)) for k in capture_flags