File size: 336 Bytes
e7fb518 94da716 e7fb518 |
1 2 3 4 5 6 7 8 9 10 11 |
# from .agent import DeepSpeedAgent
from .openlamm import LAMMPEFTModel
# def load_model(args):
# agent_name = args['models'][args['model']]['agent_name']
# model_name = args['models'][args['model']]['model_name']
# model = globals()[model_name](**args)
# agent = globals()[agent_name](model, args)
# return agent
|