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 23 Packages Projects Releases Wiki Activity

finally

Browse Source
This commit is contained in:
Arthur 2025-06-30 14:38:25 +02:00
parent 124cd82968
commit 4a14287a60
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1011,10 +1011,9 @@ def check_model_inputs(func):
def make_capture_fn(key, index):
def capture_fn(module, input, output):
if len(output) == 0:
if not isinstance(output, tuple):
collected_outputs[key] += (output,)
elif output[index] is not None:
print(key, module.__class__, output[index].shape, output[0].shape)
collected_outputs[key] += (output[index],)
return capture_fn