Function test_list_avail_args

SwissArmyTransformer/tests/test_base_model.py:4–9

()

import torch

def test_list_avail_args():
    a = BaseModel.list_avail_args()
    assert a.parse_args([]).num_layers == 24
    from sat.model import GLMModel
    a = GLMModel.list_avail_args()
    assert hasattr(a.parse_args([]), 'gpt_infill_prob')
    
def test_model_get_args():
    from sat.model import GLMModel