Include XLM-RoBERTa in the list of supported models

TheJKM · Aug 31, 2023 · 5c88c27 · 5c88c27
1 parent b4f2861
commit 5c88c27
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -10,7 +10,7 @@ The following model types are currently supported:
 
 * Encoder-decoder models: Transformer base/big, M2M-100, NLLB, BART, mBART, Pegasus, T5, Whisper
 * Decoder-only models: GPT-2, GPT-J, GPT-NeoX, OPT, BLOOM, MPT, Llama, CodeGen, GPTBigCode, Falcon
-* Encoder-only models: BERT, DistilBERT
+* Encoder-only models: BERT, DistilBERT, XLM-RoBERTa
 
 Compatible models should be first converted into an optimized model format. The library includes converters for multiple frameworks:
 

diff --git a/docs/guides/transformers.md b/docs/guides/transformers.md
@@ -22,6 +22,7 @@ CTranslate2 supports selected models from Hugging Face's [Transformers](https://
 * Pegasus
 * T5
 * Whisper
+* XLM-RoBERTa
 
 The converter takes as argument the pretrained model name or the path to a model directory: