From 517151bce4bc786020aa09877e63360b831aa52d Mon Sep 17 00:00:00 2001
From: guillaume-be <guillaume.becquin@gmail.com>
Date: Mon, 21 Dec 2020 10:42:11 +0100
Subject: [PATCH] Updated README

---
 README.md  | 1 +
 src/lib.rs | 1 +
 2 files changed, 2 insertions(+)

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| | | |✅ | | | |