- Notifications
You must be signed in to change notification settings - Fork86
DistilBERT / GPT-2 for on-device inference thanks to TensorFlow Lite with Android demo apps
License
huggingface/tflite-android-transformers
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
Convert Transformers modelsimported from the🤗 Transformers libraryand use them on Android. You can also check out ourswift-coreml-transformers repoif you're looking for Transformers on iOS.
Demo of theDistilBERT model(97% of BERT’s performance on GLUE) fine-tuned for Question answering on the SQuAD dataset.It provides 48 passages from the dataset for users to choose from.
On-device text generation usingGPT-2 orDistilGPT2 (same distillation process than DistilBERT, 2x faster and 33% smaller than GPT-2)
Example scripts used to convert the models are available in themodels_generation directory.Please note that they require the nightly version of TensorFlow and might thus be unstable.
About
DistilBERT / GPT-2 for on-device inference thanks to TensorFlow Lite with Android demo apps
Topics
Resources
License
Uh oh!
There was an error while loading.Please reload this page.
Stars
Watchers
Forks
Releases
Packages0
Uh oh!
There was an error while loading.Please reload this page.

