Update README.md
Browse files
README.md
CHANGED
|
@@ -139,7 +139,7 @@ model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
|
|
| 139 |
|
| 140 |
generated_ids = model.generate(
|
| 141 |
**model_inputs,
|
| 142 |
-
max_new_tokens=
|
| 143 |
temperature=0.7
|
| 144 |
)
|
| 145 |
generated_ids = [
|
|
|
|
| 139 |
|
| 140 |
generated_ids = model.generate(
|
| 141 |
**model_inputs,
|
| 142 |
+
max_new_tokens=2048,
|
| 143 |
temperature=0.7
|
| 144 |
)
|
| 145 |
generated_ids = [
|