mirror of
https://github.com/deepseek-ai/DeepSeek-V3.git
synced 2025-02-23 14:18:57 -05:00
18 lines
634 B
Python
18 lines
634 B
Python
from argparse import ArgumentParser
|
|
|
|
class Parser():
|
|
def __init__(self, parser = ArgumentParser(), arg_list = []):
|
|
self.parser = parser
|
|
self.arg_list = arg_list
|
|
def apply_args(self):
|
|
for arg in self.arg_list: self.parser.add_argument(*arg)
|
|
return self
|
|
def assert_model_parallel(self):
|
|
assert self.return_args.n_experts % self.return_args().model_parallel == 0
|
|
return self
|
|
def assert_interactive():
|
|
assert self.return_args().input_file or self.return_args().interactive
|
|
return self
|
|
def return_args(self):
|
|
return self.parser.parse_args()
|
|
|