Upload generate_prompt.py with huggingface_hub
Browse files- generate_prompt.py +2 -2
generate_prompt.py
CHANGED
|
@@ -77,8 +77,8 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
|
|
| 77 |
],
|
| 78 |
"result":""
|
| 79 |
}
|
| 80 |
-
idx_real = batch_idx * batch_size + idx
|
| 81 |
-
messages =
|
| 82 |
save_['content'][0]['image'] = messages['content'][0]['image']
|
| 83 |
save_['content'][1]['text'] = messages['content'][1]['text']
|
| 84 |
|
|
|
|
| 77 |
],
|
| 78 |
"result":""
|
| 79 |
}
|
| 80 |
+
#idx_real = batch_idx * batch_size + idx
|
| 81 |
+
messages = batch[idx]
|
| 82 |
save_['content'][0]['image'] = messages['content'][0]['image']
|
| 83 |
save_['content'][1]['text'] = messages['content'][1]['text']
|
| 84 |
|