Upload infer_.py with huggingface_hub

infer_.py

@@ -51,7 +51,7 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
     input_text_list = []
     data_list = []
     save_data = []
-    
+    sd_ans = []
     # while True:
     for idx, i in enumerate(batch):
         save_ =  {
@@ -95,6 +95,7 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
         save_['content'][1]['image'] = image_path
         save_['content'][2]['text'] = question
         save_['answer'] = answer
+        sd_ans.append(answer)
         data_list.append(messages)
         save_data.append(save_)
 
@@ -119,8 +120,9 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
     output_text = processor.batch_decode(
         generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False
     )
-    for x in output_text:
-        print(x)
+    for idx,x in enumerate(output_text):
+        print('model_output: ',x)
+        print('standard_output: ',sd_ans[idx])
         
     save_["answer"] = output_text
     if output_text == answer: