Views
No views yet
| Model | #params | Arch. | Training data |
|---|---|---|---|
camembert-base | 110M | Base | OSCAR (138 GB of text) |
camembert/camembert-large | 335M | Large | CCNet (135 GB of text) |
camembert/camembert-base-ccnet | 110M | Base | CCNet (135 GB of text) |
camembert/camembert-base-wikipedia-4gb | 110M | Base | Wikipedia (4 GB of text) |
camembert/camembert-base-oscar-4gb | 110M | Base | Subsample of OSCAR (4 GB of text) |
camembert/camembert-base-ccnet-4gb | 110M | Base | Subsample of CCNet (4 GB of text) |
1from transformers import CamembertModel, CamembertTokenizer
2
3# You can replace "camembert-base" with any other model from the table, e.g. "camembert/camembert-large".
4tokenizer = CamembertTokenizer.from_pretrained("camembert/camembert-base-ccnet-4gb")
5camembert = CamembertModel.from_pretrained("camembert/camembert-base-ccnet-4gb")
6
7camembert.eval() # disable dropout (or leave in train mode to finetune)
81from transformers import pipeline
2
3camembert_fill_mask = pipeline("fill-mask", model="camembert/camembert-base-ccnet-4gb", tokenizer="camembert/camembert-base-ccnet-4gb")
4results = camembert_fill_mask("Le camembert est-il <mask> ?")
5# results
6#[{'sequence': '<s> Le camembert est-il sain?</s>', 'score': 0.07001790404319763, 'token': 10286},
7#{'sequence': '<s> Le camembert est-il français?</s>', 'score': 0.057594332844018936, 'token': 384},
8#{'sequence': '<s> Le camembert est-il bon?</s>', 'score': 0.04098724573850632, 'token': 305},
9#{'sequence': '<s> Le camembert est-il périmé?</s>', 'score': 0.03486393392086029, 'token': 30862},
10#{'sequence': '<s> Le camembert est-il cher?</s>', 'score': 0.021535946056246758, 'token': 1604}]
111import torch
2# Tokenize in sub-words with SentencePiece
3tokenized_sentence = tokenizer.tokenize("J'aime le camembert !")
4# ['▁J', "'", 'aime', '▁le', '▁ca', 'member', 't', '▁!']
5
6# 1-hot encode and add special starting and end tokens
7encoded_sentence = tokenizer.encode(tokenized_sentence)
8# [5, 133, 22, 1250, 16, 12034, 14324, 81, 76, 6]
9# NB: Can be done in one step : tokenize.encode("J'aime le camembert !")
10
11# Feed tokens to Camembert as a torch tensor (batch dim 1)
12encoded_sentence = torch.tensor(encoded_sentence).unsqueeze(0)
13embeddings, _ = camembert(encoded_sentence)
14# embeddings.detach()
15# embeddings.size torch.Size([1, 10, 768])
16#tensor([[[ 0.0331, 0.0095, -0.2776, ..., 0.2875, -0.0827, -0.2467],
17# [-0.1348, 0.0478, -0.5409, ..., 0.8330, 0.0467, 0.0662],
18# [ 0.0920, -0.0264, 0.0177, ..., 0.1112, 0.0108, -0.1123],
19# ...,1from transformers import CamembertConfig
2# (Need to reload the model with new config)
3config = CamembertConfig.from_pretrained("camembert/camembert-base-ccnet-4gb", output_hidden_states=True)
4camembert = CamembertModel.from_pretrained("camembert/camembert-base-ccnet-4gb", config=config)
5
6embeddings, _, all_layer_embeddings = camembert(encoded_sentence)
7# all_layer_embeddings list of len(all_layer_embeddings) == 13 (input embedding layer + 12 self attention layers)
8all_layer_embeddings[5]
9# layer 5 contextual embedding : size torch.Size([1, 10, 768])
10#tensor([[[-0.0144, 0.1855, 0.4895, ..., -0.1537, 0.0107, -0.2293],
11# [-0.6664, -0.0880, -0.1539, ..., 0.3635, 0.4047, 0.1258],
12# [ 0.0511, 0.0540, 0.2545, ..., 0.0709, -0.0288, -0.0779],
13# ...,1@inproceedings{martin2020camembert,
2 title={CamemBERT: a Tasty French Language Model},
3 author={Martin, Louis and Muller, Benjamin and Su{\'a}rez, Pedro Javier Ortiz and Dupont, Yoann and Romary, Laurent and de la Clergerie, {\'E}ric Villemonte and Seddah, Djam{\'e} and Sagot, Beno{\^\i}t},
4 booktitle={Proceedings of the 58th Annual Meeting of the Association for Computational Linguistics},
5 year={2020}
6}