Update README.md

README.md

@@ -64,7 +64,7 @@ input = "明月松间照，\n\n->\n\n"
 inputs = tokenizer(input, return_tensors='pt')
 inputs = inputs.to('cuda:0')
 pred = model.generate(**inputs,
-                        max_new_tokens=512,
+                        max_new_tokens=256,
                         do_sample=True,
                         temperature=0.3,
                         top_k=50,