PR #17961 add ONNX support for BLOOM

add ONNX support for BLOOM #17961

sgugger merged 7 commits into huggingface:main from NouamaneTazi:main

add onnx support for BLOOM

20097108

lewtun approved these changes on 2022-06-30

michaelbenayoun approved these changes on 2022-06-30

use TYPE_CHECKING for type annotations

c15f12a9

fix past_shape for bloom (different from gpt2)

fb1b76c3

use logical_or instead of `+` for onnx support

0f9199d5

NouamaneTazi requested a review from

michaelbenayoun 3 years ago

michaelbenayoun approved these changes on 2022-07-01

bigger `atol_for_validation` for larger bloom models

cf43ce2d

copied -> taken because it's no longer an exact copy

a6fb1be6

sgugger commented on 2022-07-01

remove "copied from" comment

71ecf5dd

sgugger approved these changes on 2022-07-01

sgugger merged b68d408f into main 3 years ago

Reviewers

michaelbenayoun

lewtun

sgugger

Assignees

No one assigned

Labels

None yet

Milestone

No milestone