Upload generate_prompt.py with huggingface_hub
Browse files- generate_prompt.py +1 -1
generate_prompt.py
CHANGED
|
@@ -60,7 +60,7 @@ json_path = args.prompt_path
|
|
| 60 |
|
| 61 |
for batch_idx in tqdm(range(begin, end, batch_size)):
|
| 62 |
batch = data[batch_idx:min(batch_idx+batch_size, end)]
|
| 63 |
-
|
| 64 |
image_list = []
|
| 65 |
input_text_list = []
|
| 66 |
data_list = []
|
|
|
|
| 60 |
|
| 61 |
for batch_idx in tqdm(range(begin, end, batch_size)):
|
| 62 |
batch = data[batch_idx:min(batch_idx+batch_size, end)]
|
| 63 |
+
print(len(batch))
|
| 64 |
image_list = []
|
| 65 |
input_text_list = []
|
| 66 |
data_list = []
|