Update README.md
This commit is contained in:
parent
04ae2a3c5a
commit
07a3b7a7c9
|
@ -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!
|
Now we are ready to try out how the model works as a chatting partner!
|
||||||
|
|
||||||
```python
|
```python
|
||||||
from transformers import AutoTokenizer,AutoModel
|
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
||||||
tokenizer = AutoTokenizer.from_pretrained("microsoft/GODEL-v1_1-large-seq2seq")
|
tokenizer = AutoTokenizer.from_pretrained("microsoft/GODEL-v1_1-large-seq2seq")
|
||||||
model = AutoModelForSeq2SeqLM.from_pretrained("microsoft/GODEL-v1_1-large-seq2seq")
|
model = AutoModelForSeq2SeqLM.from_pretrained("microsoft/GODEL-v1_1-large-seq2seq")
|
||||||
def generate(instruction, knowledge, dialog):
|
def generate(instruction, knowledge, dialog):
|
||||||
|
|
Loading…
Reference in New Issue