go-inoue commited on
Commit
a973e49
1 Parent(s): bc15a32

Update model name

Browse files
Files changed (1) hide show
  1. README.md +14 -14
README.md CHANGED
@@ -15,18 +15,18 @@ We release pre-trained language models for Modern Standard Arabic (MSA), dialect
15
  We also provide additional models that are pre-trained on a scaled-down set of the MSA variant (half, quarter, eighth, and sixteenth).
16
  The details are described in the paper *"[The Interplay of Variant, Size, and Task Type in Arabic Pre-trained Language Models](https://arxiv.org/abs/2103.06678)."*
17
 
18
- This model card describes **CAMeLBERT-MSA** (`bert-base-camelbert-msa`), a model pre-trained on the entire MSA dataset.
19
 
20
  ||Model|Variant|Size|#Word|
21
  |-|-|:-:|-:|-:|
22
- ||`bert-base-camelbert-mix`|CA,DA,MSA|167GB|17.3B|
23
- ||`bert-base-camelbert-ca`|CA|6GB|847M|
24
- ||`bert-base-camelbert-da`|DA|54GB|5.8B|
25
- |✔|`bert-base-camelbert-msa`|MSA|107GB|12.6B|
26
- ||`bert-base-camelbert-msa-half`|MSA|53GB|6.3B|
27
- ||`bert-base-camelbert-msa-quarter`|MSA|27GB|3.1B|
28
- ||`bert-base-camelbert-msa-eighth`|MSA|14GB|1.6B|
29
- ||`bert-base-camelbert-msa-sixteenth`|MSA|6GB|746M|
30
 
31
  ## Intended uses
32
  You can use the released model for either masked language modeling or next sentence prediction.
@@ -37,7 +37,7 @@ We release our fine-tuninig code [here](https://github.com/CAMeL-Lab/CAMeLBERT).
37
  You can use this model directly with a pipeline for masked language modeling:
38
  ```python
39
  >>> from transformers import pipeline
40
- >>> unmasker = pipeline('fill-mask', model='CAMeL-Lab/bert-base-camelbert-msa')
41
  >>> unmasker("الهدف من الحياة هو [MASK] .")
42
  [{'sequence': '[CLS] الهدف من الحياة هو العمل. [SEP]',
43
  'score': 0.08507660031318665,
@@ -63,8 +63,8 @@ You can use this model directly with a pipeline for masked language modeling:
63
  Here is how to use this model to get the features of a given text in PyTorch:
64
  ```python
65
  from transformers import AutoTokenizer, AutoModel
66
- tokenizer = AutoTokenizer.from_pretrained('CAMeL-Lab/bert-base-camelbert-msa')
67
- model = AutoModel.from_pretrained('CAMeL-Lab/bert-base-camelbert-msa')
68
  text = "مرحبا يا عالم."
69
  encoded_input = tokenizer(text, return_tensors='pt')
70
  output = model(**encoded_input)
@@ -73,8 +73,8 @@ output = model(**encoded_input)
73
  and in TensorFlow:
74
  ```python
75
  from transformers import AutoTokenizer, TFAutoModel
76
- tokenizer = AutoTokenizer.from_pretrained('CAMeL-Lab/bert-base-camelbert-msa')
77
- model = TFAutoModel.from_pretrained('CAMeL-Lab/bert-base-camelbert-msa')
78
  text = "مرحبا يا عالم."
79
  encoded_input = tokenizer(text, return_tensors='tf')
80
  output = model(encoded_input)
 
15
  We also provide additional models that are pre-trained on a scaled-down set of the MSA variant (half, quarter, eighth, and sixteenth).
16
  The details are described in the paper *"[The Interplay of Variant, Size, and Task Type in Arabic Pre-trained Language Models](https://arxiv.org/abs/2103.06678)."*
17
 
18
+ This model card describes **CAMeLBERT-MSA** (`bert-base-arabic-camelbert-msa`), a model pre-trained on the entire MSA dataset.
19
 
20
  ||Model|Variant|Size|#Word|
21
  |-|-|:-:|-:|-:|
22
+ ||`bert-base-arabic-camelbert-mix`|CA,DA,MSA|167GB|17.3B|
23
+ ||`bert-base-arabic-camelbert-ca`|CA|6GB|847M|
24
+ ||`bert-base-arabic-camelbert-da`|DA|54GB|5.8B|
25
+ |✔|`bert-base-arabic-camelbert-msa`|MSA|107GB|12.6B|
26
+ ||`bert-base-arabic-camelbert-msa-half`|MSA|53GB|6.3B|
27
+ ||`bert-base-arabic-camelbert-msa-quarter`|MSA|27GB|3.1B|
28
+ ||`bert-base-arabic-camelbert-msa-eighth`|MSA|14GB|1.6B|
29
+ ||`bert-base-arabic-camelbert-msa-sixteenth`|MSA|6GB|746M|
30
 
31
  ## Intended uses
32
  You can use the released model for either masked language modeling or next sentence prediction.
 
37
  You can use this model directly with a pipeline for masked language modeling:
38
  ```python
39
  >>> from transformers import pipeline
40
+ >>> unmasker = pipeline('fill-mask', model='CAMeL-Lab/bert-base-arabic-camelbert-msa')
41
  >>> unmasker("الهدف من الحياة هو [MASK] .")
42
  [{'sequence': '[CLS] الهدف من الحياة هو العمل. [SEP]',
43
  'score': 0.08507660031318665,
 
63
  Here is how to use this model to get the features of a given text in PyTorch:
64
  ```python
65
  from transformers import AutoTokenizer, AutoModel
66
+ tokenizer = AutoTokenizer.from_pretrained('CAMeL-Lab/bert-base-arabic-camelbert-msa')
67
+ model = AutoModel.from_pretrained('CAMeL-Lab/bert-base-arabic-camelbert-msa')
68
  text = "مرحبا يا عالم."
69
  encoded_input = tokenizer(text, return_tensors='pt')
70
  output = model(**encoded_input)
 
73
  and in TensorFlow:
74
  ```python
75
  from transformers import AutoTokenizer, TFAutoModel
76
+ tokenizer = AutoTokenizer.from_pretrained('CAMeL-Lab/bert-base-arabic-camelbert-msa')
77
+ model = TFAutoModel.from_pretrained('CAMeL-Lab/bert-base-arabic-camelbert-msa')
78
  text = "مرحبا يا عالم."
79
  encoded_input = tokenizer(text, return_tensors='tf')
80
  output = model(encoded_input)