Swift Core ML 3 implementations of GPT-2, DistilGPT-2, BERT, and DistilBERT for Question answering. Other Transformers coming soon!
APACHE-2.0 License
This repository contains:
BasicTokenizer
and WordpieceTokenizer
) and SQuAD dataset parsing utilities.transformers
repo) to CoreML models.Unleash the full power of text generation with GPT-2 on device!!
The BERTSQUADFP16
Core ML model was packaged by Apple and is linked from the main ML models page. It was demoed at WWDC 2019 as part of the Core ML 3 launch.
The DistilBERT
Core ML models were converted from 🤗/transformers
exports using the scripts in this repo.
Apple demo at WWDC 2019
full video here
We use git-lfs
to store large model files and it is required to obtain some of the files the app needs to run.
See how to install git-lfs
on the installation page