Update moa/main.py

moa/main.py

@@ -2,11 +2,11 @@ from agent import MOAgent
 
 # Configure agent
 layer_agent_config = {
-    'layer_agent_1' : {'system_prompt': "Think through your response with step by step {helper_response}", 'model_name': 'llama3-8b-8192'},
-    'layer_agent_2' : {'system_prompt': "Respond with a thought and then your response to the question {helper_response}", 'model_name': 'gemma-7b-it'},
-    'layer_agent_3' : {'model_name': 'llama3-8b-8192'},
-    'layer_agent_4' : {'model_name': 'gemma-7b-it'},
-    'layer_agent_5' : {'model_name': 'llama3-8b-8192'},
+    'layer_agent_1' : {'system_prompt': "Think through your response with step by step {helper_response}", 'model_name': 'llama-3.2-3b-preview'},
+    'layer_agent_2' : {'system_prompt': "Respond with a thought and then your response to the question {helper_response}", 'model_name': 'llama3-70b-8192'},
+    'layer_agent_3' : {'model_name': 'llama3-70b-8192'},
+    'layer_agent_4' : {'model_name': 'llama-3.2-1b-preview'},
+    'layer_agent_5' : {'model_name': 'llama-3.2-3b-preview'},
 }
 agent = MOAgent.from_config(
     main_model='mixtral-8x7b-32768',