Skip to content

GitLab

Unverified Commit c475eca9 authored by Scruel Tao's avatar Scruel Tao Committed by GitHub
Browse files

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

Fix copy/paste error msg typo
parent 27c79a0f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/utils/generic.py
View file @ c475eca9
...@@ -660,7 +660,7 @@ def tensor_size(array): ...@@ -660,7 +660,7 @@ def tensor_size(array):
elif is_jax_tensor(array): elif is_jax_tensor(array):
return array.size return array.size
else: 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): def add_model_info_to_auto_map(auto_map, repo_id):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment