riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-03 03:31:05 +06:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

enforce original size to be a list (#33564)

Browse Source 
* enforce original size to be a list

* formatting

* apply datatype change to unpad_image in llava_next
This commit is contained in:
Dominik Niedziela 2024-09-18 17:38:31 +02:00 committed by GitHub
parent 8efc06ee18
commit 9db963aeed
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/transformers/models/llava_next/modeling_llava_next.py
View File

@ -123,6 +123,12 @@ def unpad_image(tensor, original_size):
Returns:
`torch.Tensor`: The unpadded image tensor.
"""
if not isinstance(original_size, (list, tuple)):
if not isinstance(original_size, (torch.Tensor, np.ndarray)):
raise TypeError(
f"image_size invalid type: {type(original_size)} not valid, should be either list, tuple, np.ndarray or tensor"
)
original_size = original_size.tolist()
original_height, original_width = original_size
current_height, current_width = tensor.shape[1:]

6
src/transformers/models/llava_onevision/modeling_llava_onevision.py
View File

@ -124,6 +124,12 @@ def unpad_image(tensor, original_size):
Returns:
`torch.Tensor`: The unpadded image tensor.
"""
if not isinstance(original_size, (list, tuple)):
if not isinstance(original_size, (torch.Tensor, np.ndarray)):
raise TypeError(
f"image_size invalid type: {type(original_size)} not valid, should be either list, tuple, np.ndarray or tensor"
)
original_size = original_size.tolist()
original_height, original_width = original_size
current_height, current_width = tensor.shape[1:]