c-my/transformers
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
2,957 Commits 171 Branches 49 Tags
ebba9e929dcea98e65d1f0e9c535d38370ee24b1
Commit Graph
1 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
Martin Schrimpf 90d3b787f6 move rp_bucket to relative_attention_bias' device 
otherwise, `rp_bucket` will always be on cpu and fail if `self.relative_attention_bias` is on cuda
 2020-01-10 15:09:10 -05:00