accelerate
take `torch.nn.Module` model into account when moving to device
#3167
Merged

take `torch.nn.Module` model into account when moving to device #3167

muellerzr merged 2 commits into huggingface:main from faaany:regression
faaany
faaany faaany changed the title bug fix take `torch.nn.Module` model into account when moving to device 1 year ago
muellerzr
muellerzr commented on 2024-10-14
HuggingFaceDocBuilderDev
faaany bug fix
7071c1e9
faaany faaany marked this pull request as ready for review 1 year ago
faaany update code
6e0d3004
dvrogozh
faaany
muellerzr
muellerzr approved these changes on 2024-10-31
muellerzr muellerzr merged 87732a4c into main 1 year ago
dvrogozh
faaany faaany deleted the regression branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone