()
| 36 | a = model(input_ids=x, position_ids=x, attention_mask=None) |
| 37 | |
| 38 | def test_full_mode_inference(): |
| 39 | args = get_args(['--zero-stage', '1']) |
| 40 | from sat.model import RobertaModel, AutoModel |
| 41 | model, args1 = RobertaModel.from_pretrained('roberta-base') |
| 42 | x = torch.tensor([[1,2,3]], device='cuda') |
| 43 | a = model(input_ids=x, position_ids=x, attention_mask=None) |
| 44 | |
| 45 | if __name__ == '__main__': |
| 46 | # test_model_inference() |
nothing calls this directly
no test coverage detected