transformers
Create the arange tensor on device for enabling CUDA-Graph for Clip Encoder
#19503
Merged
Merged
Create the arange tensor on device for enabling CUDA-Graph for Clip Encoder #19503
LysandreJik
merged
2 commits into
huggingface:main
from
RezaYazdaniAminabadi:fix-clip-transformer
create the arange tensor on device for enabling CUDA-Graph at higher-…
4edfce84
stas00
approved these changes
on 2022-10-12
sync
5960a643
LysandreJik
approved these changes
on 2022-10-12
Login to write a write a comment.
Login via GitHub