razent commited on
Commit
1c12345
1 Parent(s): a252ac1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -13
README.md CHANGED
@@ -22,19 +22,6 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
22
  tokenizer = AutoTokenizer.from_pretrained("VietAI/vit5-base")
23
  model = AutoModelForSeq2SeqLM.from_pretrained("VietAI/vit5-base")
24
  model.cuda()
25
-
26
- 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."
27
- text = sentence
28
- encoding = tokenizer(text, return_tensors="pt")
29
- input_ids, attention_masks = encoding["input_ids"].to("cuda"), encoding["attention_mask"].to("cuda")
30
- outputs = model.generate(
31
- input_ids=input_ids, attention_mask=attention_masks,
32
- max_length=256,
33
- early_stopping=True
34
- )
35
- for output in outputs:
36
- line = tokenizer.decode(output, skip_special_tokens=True, clean_up_tokenization_spaces=True)
37
- print(line)
38
  ```
39
 
40
  ## Citation
 
22
  tokenizer = AutoTokenizer.from_pretrained("VietAI/vit5-base")
23
  model = AutoModelForSeq2SeqLM.from_pretrained("VietAI/vit5-base")
24
  model.cuda()
 
 
 
 
 
 
 
 
 
 
 
 
 
25
  ```
26
 
27
  ## Citation