Wendy-Fly commited on
Commit
f56b328
·
verified ·
1 Parent(s): 5b4d60c

Upload infer_.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. infer_.py +2 -1
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