oluwatosin adewumi
commited on
Commit
•
f5c5a53
1
Parent(s):
1f69796
Wolof model modified.
Browse files
README.md
CHANGED
@@ -43,8 +43,8 @@ Now we are ready to try out how the model works as a chatting partner!
|
|
43 |
```python
|
44 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
45 |
import torch
|
46 |
-
tokenizer = AutoTokenizer.from_pretrained("tosin/
|
47 |
-
model = AutoModelForCausalLM.from_pretrained("tosin/
|
48 |
# Let's chat for 5 lines
|
49 |
for step in range(5):
|
50 |
# encode the new user input, add the eos_token and return a tensor in Pytorch
|
|
|
43 |
```python
|
44 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
45 |
import torch
|
46 |
+
tokenizer = AutoTokenizer.from_pretrained("tosin/dialogpt_afriwoz_wolof")
|
47 |
+
model = AutoModelForCausalLM.from_pretrained("tosin/dialogpt_afriwoz_wolof")
|
48 |
# Let's chat for 5 lines
|
49 |
for step in range(5):
|
50 |
# encode the new user input, add the eos_token and return a tensor in Pytorch
|