Suraj Patil e92190c0f8 Fix Flax params dtype (#13098) ... * fix inits * fix embed dtype * fix embed dtype * add test to check default dtype * quality * add type conversion methods for flax models * more robust casting * cast sinusoidal positions * update pegasus * update albert * update test * make sure dtype is passed to every module * style * fix electra dense * fix t5 * quality * add more tests * better name * use the dtype for lm head computation * fix albert * style * fix albert embed dtype * more tests * fix vision enc-dec * cleanup * fix embed dtype pegasus * fix default param test * doc * update template * fix final_logits_bias dtype * Apply suggestions from code review Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * fix doc * fix doc * add detailed docstring for dtype parameter * remove un-necessary import Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>		2021-11-11 14:45:20 +05:30
..
adding_a_new_example_script	Update namespaces inside torch.utils.data to the latest. (#13167)	2021-08-19 14:29:51 +02:00
adding_a_new_model	Fix Flax params dtype (#13098)	2021-11-11 14:45:20 +05:30