transformers
105c3a48 - Support extraction of both train and eval XLA graphs (#20492)

Commit

3 years ago

Support extraction of both train and eval XLA graphs (#20492) Neuron supports extraction of XLA graphs for compilation. However, when both do_train and do_eval options are enabled, sizes returned by tensor operator can be 0. To avoid INVALID_ARGUMENT error, we use inequality in the check whether a tensor needs padding or not.

References

#19449 - [WIP] Fix weights initialization of several vision models

#20492 - Support extraction of both train and eval XLA graphs

#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

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

#41212 - Add EoMT with DINOv3 backbone

#62 - Add initial DEIMv2 model implementation

Author

jeffhataws

Parents

b75255cd

transformers 105c3a48 - Support extraction of both train and eval XLA graphs (#20492)

transformers
105c3a48 - Support extraction of both train and eval XLA graphs (#20492)