AI Briefing
KO

Hugging Face Improves TensorFlow Model Performance

·2021.01.26 09:00

Key point

The performance of TensorFlow models in Hugging Face Transformers has been improved, becoming up to 10% faster than Google's implementation.

Details

The Hugging Face team made updates to make TensorFlow models in the Transformers library more robust and faster. This improvement focuses on enhancing the computational performance of BERT, RoBERTa, ELECTRA, MPNet models and enabling efficient deployment via TensorFlow Serving.

Key Performance Improvements

  • Improved Computational Performance: For the BERT model, performance was recorded at up to about 10% faster than Google's official implementation. This is about 2x faster compared to the previous version (v4.1.1).
  • General Benefits: Performance improvements appear across Graph/Eager mode, TF Serving, and all CPU/GPU/TPU device environments.

TensorFlow Serving Deployment Support To enable efficient model deployment in production environments, SavedModel format is supported. Starting with Transformers v4.2.0, the following features were added when creating SavedModels:

  • Ability to freely change sequence length between runs
  • Ability to use all model inputs during inference
  • hidden states and attention outputs provided together as a single group

This summary was generated automatically by AI. Check the original for the author's claims and context. Copyright belongs to the original author.

Our guide explains how the AI works. Report summary errors, attribution issues, or removal requests via Contact.