moe-llava
1import os
2import json
3import argparse
4
5parser = argparse.ArgumentParser()
6parser.add_argument("--src", type=str)
7parser.add_argument("--dst", type=str)
8args = parser.parse_args()
9
10cur_result = {}
11
12for line in open(args.src):
13data = json.loads(line)
14qid = data['question_id']
15cur_result[f'v1_{qid}'] = data['text']
16
17with open(args.dst, 'w') as f:
18json.dump(cur_result, f, indent=2)
19