transformers
99d67f1a
- Improve greedy search memory usage (#32895)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Improve greedy search memory usage (#32895) Do not call torch.repeat_interleave if expand_size is 1
References
#32895 - Improve greedy search memory usage
Author
regisss
Parents
bf97d4aa
Loading