transformers
d49f8d31 - Added type hints for Pytorch Marian calls (#16200)
Commit
View On GitHub
Commit
4 years ago
Added type hints for Pytorch Marian calls (#16200) * Added type hinting for forward functions in pytorch marian * typo correction * Removed type hints on functions from BART per Suraj Patil request * fix import pb * fix typo * corrected tuple call * ran black * after fix-copies Some optional tags on primitives were removed, past_key_values in MarianForCausalLM changed from Tuple of Tuple to List * Fixing copies to roformer and pegasus Co-authored-by: Clementine Fourrier <cfourrie@inria.fr> Co-authored-by: matt <rocketknight1@gmail.com>
Author
clefourrier clefourrier
Parents
a2379b92
FAQ Terms Privacy Refunds Impressum
Loading