saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Correctly pass device arg during recursion

Browse source
This commit is contained in:
Matt 2024-12-03 18:07:28 +00:00
parent 0837c7e442
commit 9cd45c8b48
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/transformers/models/pixtral/image_processing_pixtral.py
View file

@ -67,7 +67,7 @@ class BatchMixFeature(BatchFeature):
def _recursive_to(obj, device, *args, **kwargs):
# Lists can be nested, so keep digging until we hit tensors
if isinstance(obj, list):
return [_recursive_to(o, *args, **kwargs) for o in obj]
return [_recursive_to(o, device, *args, **kwargs) for o in obj]
# We cast only floating point tensors to avoid issues with tokenizers casting `LongTensor` to `FloatTensor`
elif isinstance(obj, torch.Tensor) and torch.is_floating_point(obj):
# cast and send to device

2
src/transformers/models/pixtral/processing_pixtral.py
View file

@ -70,7 +70,7 @@ class BatchMixFeature(BatchFeature):
def _recursive_to(obj, device, *args, **kwargs):
# Lists can be nested, so keep digging until we hit tensors
if isinstance(obj, list):
return [_recursive_to(o, *args, **kwargs) for o in obj]
return [_recursive_to(o, device, *args, **kwargs) for o in obj]
# We cast only floating point tensors to avoid issues with tokenizers casting `LongTensor` to `FloatTensor`
elif isinstance(obj, torch.Tensor) and torch.is_floating_point(obj):
# cast and send to device