Update README.md
Browse files
README.md
CHANGED
@@ -49,8 +49,7 @@ inputs = tokenizer(texts, return_tensors="pt", padding=True, truncation=True, pa
|
|
49 |
# conduct text completion
|
50 |
outputs = model.generate(
|
51 |
**inputs,
|
52 |
-
|
53 |
-
max_new_tokens=512
|
54 |
)
|
55 |
generated_ids = [
|
56 |
output_ids[len(input_ids):] for input_ids, output_ids in zip(inputs["input_ids"], outputs)
|
|
|
49 |
# conduct text completion
|
50 |
outputs = model.generate(
|
51 |
**inputs,
|
52 |
+
max_new_tokens=65536,
|
|
|
53 |
)
|
54 |
generated_ids = [
|
55 |
output_ids[len(input_ids):] for input_ids, output_ids in zip(inputs["input_ids"], outputs)
|