Update eval_lmdeploy.py

eval_lmdeploy.py

@@ -10,7 +10,7 @@ def main(args):
     import os
     import time
 
-    start_time = time.time()
+    run_start_time = time.time()
 
     os.makedirs("tmp", exist_ok=True)
     os.makedirs("evals_res", exist_ok=True)
@@ -18,7 +18,7 @@ def main(args):
     # get base path for eval_file
     EVAL_FILE_BASENAME = os.path.basename(EVAL_FILE)
     MODEL_NAME_STR = "+".join(args.model.split("/"))
-    SAVED_EVAL_FILE = f"{str(start_time)}_{MODEL_NAME_STR}_{EVAL_FILE_BASENAME}_seq{args.num_seqs}_tok{args.tokens}_q{args.quant_policy}_tpp{args.top_p}_mnp{args.min_p}_tpk{args.top_k}"
+    SAVED_EVAL_FILE = f"{str(run_start_time)}_{MODEL_NAME_STR}_{EVAL_FILE_BASENAME}_seq{args.num_seqs}_tok{args.tokens}_q{args.quant_policy}_tpp{args.top_p}_mnp{args.min_p}_tpk{args.top_k}"
 
     import os
 
@@ -382,6 +382,8 @@ def main(args):
         else:
             std_outputs = std_outputs + "\nAll predictions match the reference!" + "\n"
 
+        time_taken = time.time() - run_start_time
+        std_outputs = std_outputs + f"Time taken: {time_taken:.2f} seconds" + "\n"
         print(std_outputs)
 
         # write stdoutputs to evals_res/outputs_{SAVED_EVAL_FILE}.log