Update README.md
Browse files
README.md
CHANGED
@@ -58,7 +58,7 @@ How did US states get their names?
|
|
58 |
|
59 |
### Response:''', return_tensors="pt", return_attention_mask=False)
|
60 |
|
61 |
-
outputs = model.generate(**inputs, max_length=
|
62 |
text = tokenizer.batch_decode(outputs)[0]
|
63 |
print(text)
|
64 |
```
|
|
|
58 |
|
59 |
### Response:''', return_tensors="pt", return_attention_mask=False)
|
60 |
|
61 |
+
outputs = model.generate(**inputs, max_length=2048)
|
62 |
text = tokenizer.batch_decode(outputs)[0]
|
63 |
print(text)
|
64 |
```
|