"sims/git@developer.sourcefind.cn:cnjsdfcy/simbricks.git" did not exist on "b4dcbc81635fd1df2cbf6727cdeaddb4fa41a02c"
Unverified Commit 26dc6593 authored by Vishal Singh's avatar Vishal Singh Committed by GitHub
Browse files

Update README.md (#8544)

Modified Model in Action section. The class `AutoModelWithLMHead` is deprecated so changed it to `AutoModelForSeq2SeqLM` for encoder-decoder models. Removed duplicate eos token.
parent 6c8fad4f
...@@ -51,13 +51,13 @@ The training script is a slightly modified version of [this one](https://colab.r ...@@ -51,13 +51,13 @@ The training script is a slightly modified version of [this one](https://colab.r
## Model in Action 🚀 ## Model in Action 🚀
```python ```python
from transformers import AutoModelWithLMHead, AutoTokenizer from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-squadv2") tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-squadv2")
model = AutoModelWithLMHead.from_pretrained("mrm8488/t5-base-finetuned-squadv2") model = AutoModelForSeq2SeqLM.from_pretrained("mrm8488/t5-base-finetuned-squadv2")
def get_answer(question, context): def get_answer(question, context):
input_text = "question: %s context: %s </s>" % (question, context) input_text = "question: %s context: %s" % (question, context)
features = tokenizer([input_text], return_tensors='pt') features = tokenizer([input_text], return_tensors='pt')
output = model.generate(input_ids=features['input_ids'], output = model.generate(input_ids=features['input_ids'],
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment