Update README.md

This commit is contained in:
Baolin Peng 2022-10-25 16:32:44 +00:00 committed by huggingface-web
parent 04ae2a3c5a
commit 07a3b7a7c9
1 changed files with 1 additions and 1 deletions

View File

@ -32,7 +32,7 @@ ArXiv paper: [https://arxiv.org/abs/2206.11309](https://arxiv.org/abs/2206.11309
Now we are ready to try out how the model works as a chatting partner!
```python
from transformers import AutoTokenizer,AutoModel
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("microsoft/GODEL-v1_1-large-seq2seq")
model = AutoModelForSeq2SeqLM.from_pretrained("microsoft/GODEL-v1_1-large-seq2seq")
def generate(instruction, knowledge, dialog):