albert
/

albert-large-v1

@@ -6,7 +6,7 @@ datasets:
 - wikipedia
 ---
-# ALBERT Base v1
 Pretrained model on English language using a masked language modeling (MLM) objective. It was introduced in
 [this paper](https://arxiv.org/abs/1909.11942) and first released in
@@ -36,15 +36,15 @@ classifier using the features produced by the ALBERT model as inputs.
 ALBERT is particular in that it shares its layers across its Transformer. Therefore, all layers have the same weights. Using repeating layers results in a small memory footprint, however, the computational cost remains similar to a BERT-like architecture with the same number of hidden layers as it has to iterate through the same number of (repeating) layers.
-This is the first version of the base model. Version 2 is different from version 1 due to different dropout rates, additional training data, and longer training. It has better results in nearly all downstream tasks.
 This model has the following configuration:
-- 12 repeating layers
 - 128 embedding dimension
-- 768 hidden dimension
-- 12 attention heads
-- 11M parameters
 ## Intended uses & limitations
@@ -62,7 +62,7 @@ You can use this model directly with a pipeline for masked language modeling:
 ```python
 >>> from transformers import pipeline
->>> unmasker = pipeline('fill-mask', model='albert-base-v1')
 >>> unmasker("Hello I'm a [MASK] model.")
 [
    {
@@ -102,8 +102,8 @@ Here is how to use this model to get the features of a given text in PyTorch:
 ```python
 from transformers import AlbertTokenizer, AlbertModel
-tokenizer = AlbertTokenizer.from_pretrained('albert-base-v1')
-model = AlbertModel.from_pretrained("albert-base-v1")
 text = "Replace me by any text you'd like."
 encoded_input = tokenizer(text, return_tensors='pt')
 output = model(**encoded_input)
@@ -113,8 +113,8 @@ and in TensorFlow:
 ```python
 from transformers import AlbertTokenizer, TFAlbertModel
-tokenizer = AlbertTokenizer.from_pretrained('albert-base-v1')
-model = TFAlbertModel.from_pretrained("albert-base-v1")
 text = "Replace me by any text you'd like."
 encoded_input = tokenizer(text, return_tensors='tf')
 output = model(encoded_input)
@@ -127,7 +127,7 @@ predictions:
 ```python
 >>> from transformers import pipeline
->>> unmasker = pipeline('fill-mask', model='albert-base-v1')
 >>> unmasker("The man worked as a [MASK].")
 [

 - wikipedia
 ---
+# ALBERT Large v1
 Pretrained model on English language using a masked language modeling (MLM) objective. It was introduced in
 [this paper](https://arxiv.org/abs/1909.11942) and first released in
 ALBERT is particular in that it shares its layers across its Transformer. Therefore, all layers have the same weights. Using repeating layers results in a small memory footprint, however, the computational cost remains similar to a BERT-like architecture with the same number of hidden layers as it has to iterate through the same number of (repeating) layers.
+This is the first version of the large model. Version 2 is different from version 1 due to different dropout rates, additional training data, and longer training. It has better results in nearly all downstream tasks.
 This model has the following configuration:
+- 24 repeating layers
 - 128 embedding dimension
+- 1024 hidden dimension
+- 16 attention heads
+- 17M parameters
 ## Intended uses & limitations
 ```python
 >>> from transformers import pipeline
+>>> unmasker = pipeline('fill-mask', model='albert-large-v1')
 >>> unmasker("Hello I'm a [MASK] model.")
 [
    {
 ```python
 from transformers import AlbertTokenizer, AlbertModel
+tokenizer = AlbertTokenizer.from_pretrained('albert-large-v1')
+model = AlbertModel.from_pretrained("albert-large-v1")
 text = "Replace me by any text you'd like."
 encoded_input = tokenizer(text, return_tensors='pt')
 output = model(**encoded_input)
 ```python
 from transformers import AlbertTokenizer, TFAlbertModel
+tokenizer = AlbertTokenizer.from_pretrained('albert-large-v1')
+model = TFAlbertModel.from_pretrained("albert-large-v1")
 text = "Replace me by any text you'd like."
 encoded_input = tokenizer(text, return_tensors='tf')
 output = model(encoded_input)
 ```python
 >>> from transformers import pipeline
+>>> unmasker = pipeline('fill-mask', model='albert-large-v1')
 >>> unmasker("The man worked as a [MASK].")
 [