transformers
Fix ONNX export for sequence classification head
#36332
Merged

Fix ONNX export for sequence classification head #36332

echarlaix
echarlaix set dtype to int32
e579f5df
echarlaix echarlaix requested a review from Rocketknight1 Rocketknight1 1 year ago
echarlaix echarlaix requested a review from ArthurZucker ArthurZucker 1 year ago
echarlaix fix style
eac30e68
HuggingFaceDocBuilderDev
echarlaix Merge branch 'main' into int32-dtype
28309d81
echarlaix Merge branch 'main' into int32-dtype
86bb0bf2
echarlaix
Rocketknight1
Rocketknight1 approved these changes on 2025-03-10
Rocketknight1
ArthurZucker
ArthurZucker approved these changes on 2025-03-20
Rocketknight1
Rocketknight1 Rocketknight1 merged 055afdb6 into main 1 year ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone