[model cards] fix yaml in cards (#7207)

This commit is contained in:
Stas Bekman
2020-09-17 11:11:17 -07:00
committed by GitHub
parent e643a29722
commit 9c5bcab5b0
13 changed files with 44 additions and 35 deletions

View File

@@ -25,8 +25,9 @@ def write_model_card(model_card_dir, src_lang, tgt_lang, model_name):
readme = f"""
---
language: {src_lang}, {tgt_lang}
language:
- {src_lang}
- {tgt_lang}
thumbnail:
tags:
- translation

View File

@@ -25,7 +25,9 @@ def write_model_card(model_card_dir, src_lang, tgt_lang, model_name):
readme = f"""
---
language: {src_lang}, {tgt_lang}
language:
- {src_lang}
- {tgt_lang}
thumbnail:
tags:
- translation

View File

@@ -26,14 +26,14 @@ def write_model_card(model_card_dir, src_lang, tgt_lang):
readme = f"""
---
<!-- This file has been auto-generated by src/transformers/convert_fsmt_original_pytorch_checkpoint_to_pytorch.py - DO NOT EDIT or your changes will be lost -->
language: {src_lang}, {tgt_lang}
language:
- {src_lang}
- {tgt_lang}
thumbnail:
tags:
- translation
- wmt19
- facebook
license: Apache 2.0
datasets:
- http://www.statmt.org/wmt19/ ([test-set](http://matrix.statmt.org/test_sets/newstest2019.tgz?1556572561))
@@ -69,7 +69,7 @@ mname = "facebook/wmt19-{src_lang}-{tgt_lang}"
tokenizer = FSMTTokenizer.from_pretrained(mname)
model = FSMTForConditionalGeneration.from_pretrained(mname)
input = "{texts[src_lang]}
input = "{texts[src_lang]}"
input_ids = tokenizer.encode(input, return_tensors="pt")
outputs = model.generate(input_ids)
decoded = tokenizer.decode(outputs[0], skip_special_tokens=True)