Update model/__init__.py

model/__init__.py

@@ -1,10 +1,10 @@
-from .agent import DeepSpeedAgent
+# 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
+# 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