r1ck
/

doc2query-viT5

Text2Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

r1ck commited on Feb 1, 2023

Commit

94a2460

·

1 Parent(s): a5df5dc

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -28,7 +28,7 @@ It can be used for:
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import torch
-model_name = 'doc2query/msmarco-vietnamese-mt5-base-v1'
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)

 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import torch
+model_name = 'r1ck/doc2query-viT5'
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)