1 Commits

Author	SHA1	Message	Date
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