transformers
70e454c9 - fix: XLNet: relative_positional_encoding computes on CPU every forward (#44782)

Commit

4 days ago

fix: XLNet: relative_positional_encoding computes on CPU every forward (#44782) fix: pass device to torch.arange in XLNet relative_positional_encoding

References

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

Author

JiwaniZakir

Parents

be6cf084

transformers 70e454c9 - fix: XLNet: relative_positional_encoding computes on CPU every forward (#44782)

transformers
70e454c9 - fix: XLNet: relative_positional_encoding computes on CPU every forward (#44782)