transformers
Compute the mask in-place, with less memory reads, and on CUDA on `XLNetLMHeadModel`
#23332
Merged
Merged
Compute the mask in-place, with less memory reads, and on CUDA on `XLNetLMHeadModel` #23332
amyeroberts
merged
1 commit into
huggingface:main
from
lezcano:mask
Compute the mask in-place, with less memory reads, and on CUDA on `XL…
8b54be1c
younesbelkada
approved these changes
on 2023-05-12
amyeroberts
approved these changes
on 2023-05-12
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub