diff --git a/README.md b/README.md index ab9d042..1640202 100644 --- a/README.md +++ b/README.md @@ -13,6 +13,7 @@ The following models are currently implemented: | |**Sequence classification**|**Token classification**|**Question answering**|**Text Generation**|**Summarization**|**Translation**|**Masked LM**| :-----:|:----:|:----:|:-----:|:----:|:-----:|:----:|:----: DistilBERT|✅|✅|✅| | | |✅| +MobileBERT|✅|✅|✅| | | |✅| BERT|✅|✅|✅| | | |✅| RoBERTa|✅|✅|✅| | | |✅| GPT| | | |✅ | | | | diff --git a/src/lib.rs b/src/lib.rs index 78fc5a9..e19d680 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -34,6 +34,7 @@ //! | |**Sequence classification**|**Token classification**|**Question answering**|**Text Generation**|**Summarization**|**Translation**|**Masked LM**| //! :-----:|:----:|:----:|:-----:|:----:|:-----:|:----:|:----: //! DistilBERT|✅|✅|✅| | | |✅| +//! MobileBERT|✅|✅|✅| | | |✅| //! BERT|✅|✅|✅| | | |✅| //! RoBERTa|✅|✅|✅| | | |✅| //! GPT| | | |✅ | | | |