update

examples/api_eval/eval_gemini_google.py

@@ -19,6 +19,7 @@ def get_args():
     """
     python3 eval_gemini_google.py --model_name gemini-2.5-pro --eval_result eval_math_result_gemini-2.5-pro.jsonl
     python3 eval_gemini_google.py --model_name gemini-2.5-flash --eval_result eval_math_result_gemini-2.5-flash.jsonl
+    python3 eval_gemini_google.py --model_name gemini-2.5-flash-lite-preview-06-17 --eval_result eval_math_result_gemini-2.5-flash-lite-preview-06-17.jsonl
     :return:
     """
     parser = argparse.ArgumentParser()
@@ -82,7 +83,7 @@ def main():
 
     with open(args.eval_data, "r", encoding="utf-8") as fin, open(args.eval_result, "a+", encoding="utf-8") as fout:
         for row in fin:
-            if total > 1000:
+            if total > 20:
                 break
 
             row = json.loads(row)
@@ -132,7 +133,7 @@ Your output is:
                 config=generate_content_config,
             )
             time_cost = time.time() - time_begin
-            # print(time_cost)
+            print(time_cost)
             try:
                 prediction = response.candidates[0].content.parts[0].text
             except TypeError:
@@ -158,7 +159,7 @@ Your output is:
             row_ = json.dumps(row_, ensure_ascii=False)
             fout.write(f"{row_}\n")
 
-            print(f"score: {score}")
+            # print(f"score: {score}")
 
     return