replace tgt_lang by tgt_text (#13061)

This commit is contained in:
SaulLu
2021-08-09 19:17:05 +02:00
committed by GitHub
parent a8bf2fa76e
commit 76cadb7943

View File

@@ -58,7 +58,7 @@ examples. To install :obj:`sentencepiece` run ``pip install sentencepiece``.
tokenizer = M2M100Tokenizer.from_pretrained('facebook/m2m100_418M', src_lang="en", tgt_lang="fr") tokenizer = M2M100Tokenizer.from_pretrained('facebook/m2m100_418M', src_lang="en", tgt_lang="fr")
src_text = "Life is like a box of chocolates." src_text = "Life is like a box of chocolates."
tgt_lang = "La vie est comme une boîte de chocolat." tgt_text = "La vie est comme une boîte de chocolat."
model_inputs = tokenizer(src_text, return_tensors="pt") model_inputs = tokenizer(src_text, return_tensors="pt")
with tokenizer.as_target_tokenizer(): with tokenizer.as_target_tokenizer():