(extra_args_provider)
| 38 | |
| 39 | |
| 40 | def initialize(extra_args_provider): |
| 41 | parser = argparse.ArgumentParser(add_help=False) |
| 42 | add_bminf_args(parser) |
| 43 | add_quantization_args(parser) |
| 44 | add_initialization_args(parser) |
| 45 | GLM130B.add_model_specific_args(parser) |
| 46 | extra_args_provider(parser) |
| 47 | known, args_list = parser.parse_known_args() |
| 48 | args = get_args(args_list) |
| 49 | args = argparse.Namespace(**vars(args), **vars(known)) |
| 50 | args.do_train = False |
| 51 | initialize_distributed(args) |
| 52 | return args |
| 53 | |
| 54 | |
| 55 | def initialize_model_and_tokenizer(args): |
no test coverage detected