t5 (#20340)

Commit

3 years ago

[FLAX] Add dtype to embedding for bert/bart/opt/t5 (#20340) * [FLAX] Add dtype to embedding for bert/bart/opt/t5 * Fix all copies * Add a test case

References

Author

merrymercy

Parents