transformers
d3af76df
- [Backend support] Allow `num_logits_to_keep` as Tensor + add flag (#35757)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
325 days ago
[Backend support] Allow `num_logits_to_keep` as Tensor + add flag (#35757) * support * Update modeling_utils.py * style * most models * Other models * fix-copies * tests + generation utils
References
#35757 - [Backend support] Allow `num_logits_to_keep` as Tensor and change it to `logits_to_keep` + add flag
Author
Cyrilvallez
Parents
8736e91a
Loading