riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Arthur 2025-06-30 16:49:11 +02:00
parent a74974d989
commit 3fb6b710f2
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/transformers/utils/generic.py
View File

@ -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