Update README.md
Browse files
README.md
CHANGED
@@ -7,7 +7,8 @@ license: mit
|
|
7 |
An generative AI made using [microsoft/DialoGPT-small](https://huggingface.co/microsoft/DialoGPT-small).
|
8 |
Trained on:-
|
9 |
https://www.kaggle.com/Cornell-University/movie-dialog-corpus
|
10 |
-
|
|
|
11 |
Example:-
|
12 |
```python
|
13 |
from transformers import AutoTokenizer, AutoModelWithLMHead
|
@@ -25,7 +26,7 @@ for step in range(4):
|
|
25 |
chat_history_ids = model.generate(
|
26 |
bot_input_ids, max_length=200,
|
27 |
pad_token_id=tokenizer.eos_token_id,
|
28 |
-
no_repeat_ngram_size=
|
29 |
do_sample=True,
|
30 |
top_k=100,
|
31 |
top_p=0.7,
|
|
|
7 |
An generative AI made using [microsoft/DialoGPT-small](https://huggingface.co/microsoft/DialoGPT-small).
|
8 |
Trained on:-
|
9 |
https://www.kaggle.com/Cornell-University/movie-dialog-corpus
|
10 |
+
https://www.kaggle.com/jef1056/discord-data
|
11 |
+
|
12 |
Example:-
|
13 |
```python
|
14 |
from transformers import AutoTokenizer, AutoModelWithLMHead
|
|
|
26 |
chat_history_ids = model.generate(
|
27 |
bot_input_ids, max_length=200,
|
28 |
pad_token_id=tokenizer.eos_token_id,
|
29 |
+
no_repeat_ngram_size=4,
|
30 |
do_sample=True,
|
31 |
top_k=100,
|
32 |
top_p=0.7,
|