Update README.md
Browse files
README.md
CHANGED
@@ -45,7 +45,7 @@ This is the model card of a 🤗 transformers model that has been pushed on the
|
|
45 |
```python
|
46 |
from transformers import GPT2LMHeadModel, GPT2Tokenizer
|
47 |
|
48 |
-
models_folder = "Deeokay/gpt2-
|
49 |
|
50 |
model = GPT2LMHeadModel.from_pretrained(models_folder)
|
51 |
tokenizer = GPT2Tokenizer.from_pretrained(models_folder)
|
@@ -60,10 +60,10 @@ set_seed(42)
|
|
60 |
|
61 |
sample_output = model.generate(
|
62 |
**inputs,
|
63 |
-
max_new_tokens=
|
64 |
do_sample=True,
|
65 |
-
top_k=
|
66 |
-
temperature=0.
|
67 |
)
|
68 |
|
69 |
print("Output:\n" + 80 * '-')
|
|
|
45 |
```python
|
46 |
from transformers import GPT2LMHeadModel, GPT2Tokenizer
|
47 |
|
48 |
+
models_folder = "Deeokay/gpt2-javis-stks"
|
49 |
|
50 |
model = GPT2LMHeadModel.from_pretrained(models_folder)
|
51 |
tokenizer = GPT2Tokenizer.from_pretrained(models_folder)
|
|
|
60 |
|
61 |
sample_output = model.generate(
|
62 |
**inputs,
|
63 |
+
max_new_tokens=250,
|
64 |
do_sample=True,
|
65 |
+
top_k=30,
|
66 |
+
temperature=0.7,
|
67 |
)
|
68 |
|
69 |
print("Output:\n" + 80 * '-')
|