Update README.md

README.md

@@ -35,7 +35,7 @@ input = TEMPLATE_V2.format(raw_prompt=raw_prompt)
 input_ids = tokenizer.encode(input, return_tensors='pt').cuda()
 outputs = model.generate(
     input_ids,
-    max_length=384,
+    max_new_tokens=384,
     do_sample=True,
     temperature=0.9,
     top_k=50,