Build the model.
(pre_process=True, post_process=True)
| 24 | |
| 25 | |
| 26 | def model_provider(pre_process=True, post_process=True): |
| 27 | """Build the model.""" |
| 28 | |
| 29 | print_rank_0("Building CodeGeeX model ...") |
| 30 | model = CodeGeeXModel(num_tokentypes=0, |
| 31 | parallel_output=False) |
| 32 | |
| 33 | return model |
| 34 | |
| 35 | |
| 36 | def add_code_generation_args(parser): |
nothing calls this directly
no test coverage detected