transformers
fix: XLNet: relative_positional_encoding computes on CPU every forward
#44782
Merged
Overview
Commits
1
Changes
View On GitHub

fix: XLNet: relative_positional_encoding computes on CPU every forward #44782

Rocketknight1 merged 1 commit into huggingface:main from JiwaniZakir:fix/issue-44737
JiwaniZakir
JiwaniZakir fix: pass device to torch.arange in XLNet relative_positional_encoding
d7659c10
github-actions
Rocketknight1
Rocketknight1
Rocketknight1 approved these changes on 2026-03-18
github-actions
github-actions
HuggingFaceDocBuilderDev
JiwaniZakir
Rocketknight1 Rocketknight1 merged 70e454c9 into main 15 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum