Update README.md
Browse files
README.md
CHANGED
@@ -64,7 +64,7 @@ text = "When faced with a complex problem, what should one do?"
|
|
64 |
input_ids = tokenizer(text, return_tensors="pt").to('cuda').input_ids
|
65 |
outputs = model.generate(
|
66 |
input_ids,
|
67 |
-
max_length=
|
68 |
temperature=0.6,
|
69 |
top_p=0.9,
|
70 |
repetition_penalty=1.2,
|
|
|
64 |
input_ids = tokenizer(text, return_tensors="pt").to('cuda').input_ids
|
65 |
outputs = model.generate(
|
66 |
input_ids,
|
67 |
+
max_length=2500,
|
68 |
temperature=0.6,
|
69 |
top_p=0.9,
|
70 |
repetition_penalty=1.2,
|