Downloads · 30 days
244
1% of all-time downloads
t5-small-squad11
allenai/t5-small-squad11
t5-small-squad11 is a machine learning model from allenai. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers.
SQuAD 1.1 question-answering based on T5-small. Example use:
All-time downloads
30.5K
Public
Repo size
727 MB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin242 MB · 50%
At a glance
- Library
- transformers
- Model type
- t5
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Jan 24, 2023
- SHA
- 030d60bf
Model card
From the Hugging Face model README
SQuAD 1.1 question-answering based on T5-small. Example use:
from transformers import T5Config, T5ForConditionalGeneration, T5Tokenizer
model_name = "allenai/t5-small-next-word-generator-qoogle"
tokenizer = T5Tokenizer.from_pretrained(model_name)
model = T5ForConditionalGeneration.from_pretrained(model_name)
def run_model(input_string, **generator_args):
input_ids = tokenizer.encode(input_string, return_tensors="pt")
res = model.generate(input_ids, **generator_args)
output = tokenizer.batch_decode(res, skip_special_tokens=True)
print(output)
return output
run_model("Who is the winner of 2009 olympics? \n Jack and Jill participated, but James won the games.")```
which should result in the following:
['James']
- Library
- transformers
- Type
- t5
- Languages
- en
- Created
- Mar 2, 2022
- Updated
- Jan 24, 2023