()
| 9 | assert hasattr(a.parse_args([]), 'gpt_infill_prob') |
| 10 | |
| 11 | def test_model_get_args(): |
| 12 | from sat.model import GLMModel |
| 13 | args = GLMModel.get_args() |
| 14 | assert args.num_layers == 24 |
| 15 | print(args) |
| 16 | args = GLMModel.get_args(num_layers=2) |
| 17 | assert args.num_layers == 2 |
| 18 | |
| 19 | def test_model_from_pretrained(): |
| 20 | from sat.model import RobertaModel, AutoModel |