VietAI
/

vit5-base

Question Answering

text2text-generation

text-generation-inference

Model card Files Files and versions

razent commited on Jul 25, 2022

Commit

1e7647a

·

1 Parent(s): 04d9605

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -23,7 +23,7 @@ tokenizer = AutoTokenizer.from_pretrained("VietAI/vit5-base")
 model = AutoModelForSeq2SeqLM.from_pretrained("VietAI/vit5-base")
 sentence = "VietAI là tổ chức phi lợi nhuận với sứ mệnh ươm mầm tài năng về trí tuệ nhân tạo và xây dựng một cộng đồng các chuyên gia trong lĩnh vực trí tuệ nhân tạo đẳng cấp quốc tế tại Việt Nam."
-text =  "summarize: " + sentence + " </s>"
 encoding = tokenizer.encode_plus(text, pad_to_max_length=True, return_tensors="pt")
 input_ids, attention_masks = encoding["input_ids"].to("cuda"), encoding["attention_mask"].to("cuda")
 outputs = model.generate(

 model = AutoModelForSeq2SeqLM.from_pretrained("VietAI/vit5-base")
 sentence = "VietAI là tổ chức phi lợi nhuận với sứ mệnh ươm mầm tài năng về trí tuệ nhân tạo và xây dựng một cộng đồng các chuyên gia trong lĩnh vực trí tuệ nhân tạo đẳng cấp quốc tế tại Việt Nam."
+text =  "vi: " + sentence
 encoding = tokenizer.encode_plus(text, pad_to_max_length=True, return_tensors="pt")
 input_ids, attention_masks = encoding["input_ids"].to("cuda"), encoding["attention_mask"].to("cuda")
 outputs = model.generate(