transformers
fix: XLNet: relative_positional_encoding computes on CPU every forward
#44782
Merged
Overview
Commits
1
Changes
View On GitHub
Commits
  • fix: pass device to torch.arange in XLNet relative_positional_encoding
    JiwaniZakir committed 26 days ago
FAQ Terms Privacy Refunds Impressum
Loading