transformers
Replaces calls to `.cuda` with `.to(torch_device)` in tests
#25571

Merged

Replaces calls to `.cuda` with `.to(torch_device)` in tests #25571

ArthurZucker merged 6 commits into huggingface:main from vvvm23:testing-remove-dotcuda

Replaces calls to `.cuda` with `.to(torch_device)` in tests

c3a16ec2

ArthurZucker commented on 2023-08-17

ArthurZucker commented on 2023-08-18

addressing review comments

e3451745

ArthurZucker commented on 2023-08-18

more formatting changes in Bloom test

f9c1f7ea

`make style`

150b50b2

ArthurZucker commented on 2023-08-18

Update tests/models/bloom/test_modeling_bloom.py

28f9bb72

fixes style failures

e82952e5

ArthurZucker approved these changes on 2023-08-18

ArthurZucker merged 9d7afd25 into main 2 years ago

vvvm23 deleted the testing-remove-dotcuda branch 2 years ago

Reviewers

ArthurZucker

Assignees

No one assigned

Labels

None yet

Milestone

No milestone