Upload generate_prompt.py with huggingface_hub
Browse files- generate_prompt.py +2 -2
generate_prompt.py
CHANGED
|
@@ -96,8 +96,8 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
|
|
| 96 |
|
| 97 |
for idx,x in enumerate(output_text):
|
| 98 |
idx_real = batch_idx * batch_size + idx
|
| 99 |
-
save_list[
|
| 100 |
-
save_data.append(save_list[
|
| 101 |
|
| 102 |
json_path = "image_path = '/home/zbz5349/WorkSpace/aigeeks/Qwen2.5-VL/magicbrush_dataset/gen.json'"
|
| 103 |
write_json(json_path,save_data)
|
|
|
|
| 96 |
|
| 97 |
for idx,x in enumerate(output_text):
|
| 98 |
idx_real = batch_idx * batch_size + idx
|
| 99 |
+
save_list[idx]['result'] = x
|
| 100 |
+
save_data.append(save_list[idx])
|
| 101 |
|
| 102 |
json_path = "image_path = '/home/zbz5349/WorkSpace/aigeeks/Qwen2.5-VL/magicbrush_dataset/gen.json'"
|
| 103 |
write_json(json_path,save_data)
|