NlpHUST
/

t5-en-vi-small

Text2Text Generation Transformers PyTorch JAX t5 Inference Endpoints text-generation-inference

Model card Files Files and versions Community

nhanv commited on Apr 15, 2021

Commit

d2ce7f2

•

1 Parent(s): 735b39a

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -30,7 +30,7 @@ The results on test set.
 ``` bash
 import torch
-from transformers import MT5ForConditionalGeneration, T5Tokenizer
 import torch
 if torch.cuda.is_available():
     device = torch.device("cuda")
@@ -42,7 +42,7 @@ else:
     print('No GPU available, using the CPU instead.')
     device = torch.device("cpu")
-model = MT5ForConditionalGeneration.from_pretrained("NlpHUST/t5-en-vi-small")
 tokenizer = T5Tokenizer.from_pretrained("NlpHUST/t5-en-vi-small")
 model.to(device)

 ``` bash
 import torch
+from transformers import T5ForConditionalGeneration, T5Tokenizer
 import torch
 if torch.cuda.is_available():
     device = torch.device("cuda")
     print('No GPU available, using the CPU instead.')
     device = torch.device("cpu")
+model = T5ForConditionalGeneration.from_pretrained("NlpHUST/t5-en-vi-small")
 tokenizer = T5Tokenizer.from_pretrained("NlpHUST/t5-en-vi-small")
 model.to(device)