Update README.md

README.md

@@ -36,7 +36,7 @@ from tqdm import tqdm
 import json
 
 HF_TOKEN = "your-token"
-model_name = "kky84176/llm-jp-3-13b-finetune"
+model_name = "kky84176/llm-jp-3-13b-instruct-it03"
 
 #
 bnb_config = BitsAndBytesConfig(
@@ -84,7 +84,7 @@ for data in tqdm(datasets):
       outputs = model.generate(
           tokenized_input,
           attention_mask=attention_mask,
-          max_new_tokens=100,
+          max_new_tokens=512,
           do_sample=False,
           repetition_penalty=1.2,
           pad_token_id=tokenizer.eos_token_id
@@ -95,7 +95,7 @@ for data in tqdm(datasets):
 
 # jsonl への出力
 import re
-new_model_id = "llm-jp-3-13b-finetune"
+new_model_id = "llm-jp-3-13b-instruct-it03"
 jsonl_id = re.sub(".*/", "", new_model_id)
 with open(f"./{jsonl_id}-outputs.jsonl", 'w', encoding='utf-8') as f:
     for result in results: