codemafia0000
commited on
update readme
Browse files
README.md
CHANGED
@@ -49,6 +49,9 @@ import json
|
|
49 |
|
50 |
# モデル名の指定
|
51 |
model_name = "codemafia0000/llm-jp-3-13b-finetune-LoRA-241124"
|
|
|
|
|
|
|
52 |
|
53 |
# Hugging Faceのアクセストークンを設定します。自身のトークンに置き換えてください。
|
54 |
HF_TOKEN = "YOUR_TOKEN"
|
@@ -104,7 +107,7 @@ for dt in tqdm(datasets):
|
|
104 |
results.append({"task_id": dt["task_id"], "input": input, "output": prediction})
|
105 |
|
106 |
# 結果をJSONL形式で保存
|
107 |
-
with open(
|
108 |
for result in results:
|
109 |
json.dump(result, f, ensure_ascii=False)
|
110 |
f.write('\n')
|
|
|
49 |
|
50 |
# モデル名の指定
|
51 |
model_name = "codemafia0000/llm-jp-3-13b-finetune-LoRA-241124"
|
52 |
+
# 提出用ファイル名
|
53 |
+
output_file_name = model_name.split('/')[-1] # モデル名のみ取得
|
54 |
+
output_file_name = f"{output_file_name}_output.jsonl"
|
55 |
|
56 |
# Hugging Faceのアクセストークンを設定します。自身のトークンに置き換えてください。
|
57 |
HF_TOKEN = "YOUR_TOKEN"
|
|
|
107 |
results.append({"task_id": dt["task_id"], "input": input, "output": prediction})
|
108 |
|
109 |
# 結果をJSONL形式で保存
|
110 |
+
with open(output_file_name, 'w', encoding='utf-8') as f:
|
111 |
for result in results:
|
112 |
json.dump(result, f, ensure_ascii=False)
|
113 |
f.write('\n')
|