transformers
e0bc4c73 - Add balanced strategies for device_map in from_pretrained (#18349)

Commit

3 years ago

Add balanced strategies for device_map in from_pretrained (#18349) * Add balanced strategies for device_map in from_pretrained * Add safeguards for Accelerate version * Update src/transformers/modeling_utils.py Co-authored-by: Lysandre Debut <lysandre.debut@reseau.eseo.fr> * Style Co-authored-by: Lysandre Debut <lysandre.debut@reseau.eseo.fr>

References

#18349 - Add balanced strategies for device_map in from_pretrained

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

#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

#62 - Add initial DEIMv2 model implementation

#65 - Fix RTDetrV2 sine position embedding ordering

Author

sgugger

Parents

39e76d76

transformers e0bc4c73 - Add balanced strategies for device_map in from_pretrained (#18349)

transformers
e0bc4c73 - Add balanced strategies for device_map in from_pretrained (#18349)