[model cards] fix yaml in cards (#7207)
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user