transformers
f7ca656f - [Flax] adding support for batch norm layers (#21581)

Commit

2 years ago

[Flax] adding support for batch norm layers (#21581) * [flax] adding support for batch norm layers * fixing bugs related to pt+flax integration * cleanup, batchnorm support in sharded pt to flax * support for batchnorm tests in pt+flax integration * simplifying checking batch norm layer

References

#21581 - [Flax] adding support for batch norm layers

#27720 - Add common processor tests

#29969 - [SigLIP] Add fast tokenizer

#32831 - [Docs] Update resources

#33111 - [Backbone] Remove out_features everywhere

#33174 - [Zero-shot image classification pipeline] Remove tokenizer_kwargs

#39821 - Support MetaCLIP 2

#58 - Add EoMT DINOv3 model

#59 - Fix attention mask handling in EoMT-DINOv3 converter

#41212 - Add EoMT with DINOv3 backbone

#62 - Add initial DEIMv2 model implementation

Author

Shubhamai

Parents

279008ad

transformers f7ca656f - [Flax] adding support for batch norm layers (#21581)

transformers
f7ca656f - [Flax] adding support for batch norm layers (#21581)