[FLAX] Add dtype to embedding for bert/bart/opt/t5 #20340
[FLAX] Add dtype to embedding for bert/bart/opt/t5
2369faed
Fix all copies
6d50a75e
Add a test case
77523e31
sgugger
merged
ac2f6674
into main 3 years ago
merrymercy
deleted the pr-dtype branch 3 years ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub