PR #17858 Add type hints for gptneox models

Add type hints for gptneox models #17858

Rocketknight1 merged 3 commits into huggingface:main from willtai:add-type-hints-for-gptneox-models

feat: Add type hints for GPTNeoxForCausalLM and GPTNeoXModel

f712a66f

fix: removed imported Dict type

d6964191

fix: Removed unused List import

0ee420ea

Rocketknight1 approved these changes on 2022-06-24

Rocketknight1 merged ef28a402 into main 3 years ago

donelianc commented on 2022-06-24

Reviewers

Rocketknight1

donelianc

Assignees

No one assigned

Labels

None yet

Milestone

No milestone