transformers
ef28a402 - Add type hints for gptneox models (#17858)

Commit
3 years ago
Add type hints for gptneox models (#17858) * feat: Add type hints for GPTNeoxForCausalLM and GPTNeoXModel * fix: removed imported Dict type * fix: Removed unused List import
Author
Parents
Loading