Suraj Patil d3bd9ac728 [Flax] improve large model init and loading (#16148) ... * begin do_init * add params_shape_tree * raise error if params are accessed when do_init is False * don't allow do_init=False when keys are missing * make shape tree a property * assign self._params at the end * add test for do_init * add do_init arg to all flax models * fix param setting * disbale do_init for composite models * update test * add do_init in FlaxBigBirdForMultipleChoice * better names and errors * improve test * style * add a warning when do_init=False * remove extra if * set params after _required_params * add test for from_pretrained * do_init => _do_init * chage warning to info * fix typo * add params in init_weights * add params to gpt neo init * add params to init_weights * update do_init test * Trigger CI * Apply suggestions from code review Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * update template * trigger CI * style * style * fix template Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>		2022-04-19 14:19:55 +02:00
..
adding_a_missing_tokenization_test	add a template to add missing tokenization test (#16553)	2022-04-05 10:50:22 +02:00
adding_a_new_example_script	added type hints for BART model (#16270)	2022-03-21 15:18:01 +00:00
adding_a_new_model	[Flax] improve large model init and loading (#16148)	2022-04-19 14:19:55 +02:00