transformers
d99f11e8 - ensure banned_mask and indices in same device (#23901)
Commit
View On GitHub
Commit
2 years ago
ensure banned_mask and indices in same device (#23901) * ensure banned_mask and indices in same device * ensure banned_mask and indices in same device switch the order in which indices and banned_mask are created and create banned_mask on the proper device
Author
cauyxy cauyxy
Parents
d68d6665
FAQ Terms Privacy Refunds Impressum
Loading