c-my/transformers
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
2,958 Commits 171 Branches 49 Tags
e83d9f1c1d29890dd470de74f41627630e52abdc
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