Upload infer_.py with huggingface_hub
Browse files
infer_.py
CHANGED
@@ -98,7 +98,7 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
|
|
98 |
save_['answer'] = answer
|
99 |
sd_ans.append(answer)
|
100 |
data_list.append(messages)
|
101 |
-
|
102 |
text = processor.apply_chat_template(data_list, tokenize=False, add_generation_prompt=True)
|
103 |
image_inputs, video_inputs, video_kwargs = process_vision_info(data_list, return_video_kwargs=True)
|
104 |
fps = 1
|
@@ -123,6 +123,7 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
|
|
123 |
for idx,x in enumerate(output_text):
|
124 |
if x == sd_ans[idx]:
|
125 |
correct_num = correct_num + 1
|
|
|
126 |
data_list[idx]['result'] = x
|
127 |
save_data.append(data_list[idx])
|
128 |
|
|
|
98 |
save_['answer'] = answer
|
99 |
sd_ans.append(answer)
|
100 |
data_list.append(messages)
|
101 |
+
|
102 |
text = processor.apply_chat_template(data_list, tokenize=False, add_generation_prompt=True)
|
103 |
image_inputs, video_inputs, video_kwargs = process_vision_info(data_list, return_video_kwargs=True)
|
104 |
fps = 1
|
|
|
123 |
for idx,x in enumerate(output_text):
|
124 |
if x == sd_ans[idx]:
|
125 |
correct_num = correct_num + 1
|
126 |
+
print(x,sd_ans[idx])
|
127 |
data_list[idx]['result'] = x
|
128 |
save_data.append(data_list[idx])
|
129 |
|