| # 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 | |