Upload generate_prompt.py with huggingface_hub
Browse files- generate_prompt.py +1 -1
generate_prompt.py
CHANGED
|
@@ -112,7 +112,7 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
|
|
| 112 |
print(output_text)
|
| 113 |
for idx,x in enumerate(output_text):
|
| 114 |
idx_real = batch_idx * batch_size + idx
|
| 115 |
-
save_list[idx]['result'] = x
|
| 116 |
save_data.append(save_list[idx])
|
| 117 |
if batch_idx % 4 ==0:
|
| 118 |
write_json(json_path,save_data)
|
|
|
|
| 112 |
print(output_text)
|
| 113 |
for idx,x in enumerate(output_text):
|
| 114 |
idx_real = batch_idx * batch_size + idx
|
| 115 |
+
save_list[idx][0]['result'] = x
|
| 116 |
save_data.append(save_list[idx])
|
| 117 |
if batch_idx % 4 ==0:
|
| 118 |
write_json(json_path,save_data)
|