Update README.md

README.md

@@ -22,7 +22,7 @@ tokenizer = AutoTokenizer.from_pretrained('MrBananaHuman/kogpt_6b_fp16')
 input_text = '이순신은'
 input_ids = tokenizer(input_text, return_tensors='pt').input_ids.to('cuda')
 
-output = model.generate(input_ids)
+output = model.generate(input_ids, max_length=64)
 print(tokenizer.decode(output[0]))
 
 >>> 이순신은 우리에게 무엇인가? 1. 머리말 이글은 임진왜란 당시 이순인이 보여준