2 Commits

Author	SHA1	Message	Date
Patrick von Platen	8faa822ddc	Allow to set config params directly in init (#1419) ... * fix * fix deprecated kwargs logic * add tests * finish	2022-11-25 15:07:09 +01:00
Akash Pannu	a124204490	Flax: Trickle down `norm_num_groups` (#789) ... * pass norm_num_groups param and add tests * set resnet_groups for FlaxUNetMidBlock2D * fixed docstrings * fixed typo * using is_flax_available util and created require_flax decorator	2022-10-11 20:05:10 +02:00