Update README.md
Browse files
README.md
CHANGED
@@ -136,7 +136,7 @@ generation = generator(
|
|
136 |
input_text,
|
137 |
do_sample=True,
|
138 |
top_k=10,
|
139 |
-
eos_token_id=tokenizer.eos_token_id
|
140 |
)
|
141 |
|
142 |
print(f"Result: {generation[0]['generated_text']}")
|
|
|
136 |
input_text,
|
137 |
do_sample=True,
|
138 |
top_k=10,
|
139 |
+
eos_token_id=tokenizer.eos_token_id
|
140 |
)
|
141 |
|
142 |
print(f"Result: {generation[0]['generated_text']}")
|