aint no one got time for big lists

README.md

@@ -120,8 +120,8 @@ streamer = TextStreamer(tokenizer, skip_prompt=True)
 prompt = "あなたは熱狂的なポケモンファンです。"
 chat = [{"role": "system", "content": prompt}]
 
-# Who is the most powerful Pokemon? Explain your choice.
-user_input = "最強のポケモンは誰ですか？その選択理由を説明してください。"
+# Who is the single most powerful Pokemon? Explain your choice.
+user_input = "ポケモンの中で1番強いのはどのキャラクターですか。最強の者をひとつだけ挙げて下さい。その選択理由を説明してください。"
 chat.append({"role": "user", "content": user_input})
 
 # Generate - add_generation_prompt to make sure it continues as assistant
@@ -134,9 +134,9 @@ with torch.no_grad():
     outputs = model.generate(
         inputs,
         pad_token_id=tokenizer.eos_token_id,
-        max_new_tokens=1000,
-        temperature=0.7,
-        repetition_penalty=1.05,
+        max_new_tokens=500,
+        temperature=0.5,
+        repetition_penalty=1.15,
         top_p=0.95,
         do_sample=True,
         streamer=streamer,
@@ -306,7 +306,7 @@ prompt = "あなたは熱狂的なポケモンファンです。"
 chat = [{"role": "system", "content": prompt}]
 
 # Who is the most powerful Pokemon? Explain your choice.
-user_input = "最強のポケモンは誰ですか？その選択理由を説明してください。"
+user_input = "ポケモンの中で1番強いのはどのキャラクターですか。最強の者をひとつだけ挙げて下さい。その選択理由を説明してください。"
 chat.append({"role": "user", "content": user_input})
 
 # 生成 - add_generation_promptを追加してアシスタントとして続行することを確認します
@@ -319,9 +319,9 @@ with torch.no_grad():
     outputs = model.generate(
         inputs,
         pad_token_id=tokenizer.eos_token_id,
-        max_new_tokens=1000,
-        temperature=0.7,
-        repetition_penalty=1.05,
+        max_new_tokens=500,
+        temperature=0.5,
+        repetition_penalty=1.15,
         top_p=0.95,
         do_sample=True,
         streamer=streamer,