Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Akash Pannu's avatar
    Flax: Trickle down `norm_num_groups` (#789) · a1242044
    Akash Pannu authored 
    * 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
    a1242044
vae_flax.py 29.7 KB