Pārlūkot izejas kodu

Mark required args

Aleksandr Borzunov 2 gadi atpakaļ
vecāks
revīzija
84776bff73
1 mainītis faili ar 2 papildinājumiem un 2 dzēšanām
  1. 2 2
      src/petals/cli/benchmark_inference.py

+ 2 - 2
src/petals/cli/benchmark_inference.py

@@ -15,8 +15,8 @@ logger = get_logger()
 def main():
     parser = argparse.ArgumentParser()
     parser.add_argument("--model", type=str, default="bigscience/bloom-petals")
-    parser.add_argument("-p", "--n_processes", type=int)
-    parser.add_argument("-l", "--seq_len", type=int)
+    parser.add_argument("-p", "--n_processes", type=int, required=True)
+    parser.add_argument("-l", "--seq_len", type=int, required=True)
     args = parser.parse_args()
 
     processes = [mp.Process(target=benchmark_inference, args=(i, args,)) for i in range(args.n_processes)]