Upload generate_prompt.py with huggingface_hub
Browse files- generate_prompt.py +1 -1
generate_prompt.py
CHANGED
|
@@ -110,7 +110,7 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
|
|
| 110 |
generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False
|
| 111 |
)
|
| 112 |
#print(output_text)
|
| 113 |
-
save_[0]['result'] =
|
| 114 |
save_data.append(save_)
|
| 115 |
|
| 116 |
if batch_idx % 4 ==0:
|
|
|
|
| 110 |
generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False
|
| 111 |
)
|
| 112 |
#print(output_text)
|
| 113 |
+
save_[0]['result'] = output_text
|
| 114 |
save_data.append(save_)
|
| 115 |
|
| 116 |
if batch_idx % 4 ==0:
|