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

tensor_size - fix copy/paste error msg typo (#28660)

Browse Source 
Fix copy/paste error msg typo
This commit is contained in:
Scruel Tao 2024-01-23 19:22:02 +08:00 committed by GitHub
parent 27c79a0fb4
commit c475eca9cd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -660,7 +660,7 @@ def tensor_size(array):
elif is_jax_tensor(array):
return array.size
else:
raise ValueError(f"Type not supported for expand_dims: {type(array)}.")
raise ValueError(f"Type not supported for tensor_size: {type(array)}.")
def add_model_info_to_auto_map(auto_map, repo_id):