Update moa/main.py
Browse files- moa/main.py +2 -2
moa/main.py
CHANGED
@@ -4,12 +4,12 @@ from agent import MOAgent
|
|
4 |
layer_agent_config = {
|
5 |
'layer_agent_1' : {'system_prompt': "Think through your response with step by step {helper_response}", 'model_name': 'llama-3.2-3b-preview'},
|
6 |
'layer_agent_2' : {'system_prompt': "Respond with a thought and then your response to the question {helper_response}", 'model_name': 'llama3-70b-8192'},
|
7 |
-
'layer_agent_3' : {'model_name': '
|
8 |
'layer_agent_4' : {'model_name': 'llama-3.2-1b-preview'},
|
9 |
'layer_agent_5' : {'model_name': 'llama-3.2-3b-preview'},
|
10 |
}
|
11 |
agent = MOAgent.from_config(
|
12 |
-
main_model='
|
13 |
layer_agent_config=layer_agent_config
|
14 |
)
|
15 |
|
|
|
4 |
layer_agent_config = {
|
5 |
'layer_agent_1' : {'system_prompt': "Think through your response with step by step {helper_response}", 'model_name': 'llama-3.2-3b-preview'},
|
6 |
'layer_agent_2' : {'system_prompt': "Respond with a thought and then your response to the question {helper_response}", 'model_name': 'llama3-70b-8192'},
|
7 |
+
'layer_agent_3' : {'model_name': 'llama-3.3-70b-versatile'},
|
8 |
'layer_agent_4' : {'model_name': 'llama-3.2-1b-preview'},
|
9 |
'layer_agent_5' : {'model_name': 'llama-3.2-3b-preview'},
|
10 |
}
|
11 |
agent = MOAgent.from_config(
|
12 |
+
main_model='llama-3.3-70b-versatile',
|
13 |
layer_agent_config=layer_agent_config
|
14 |
)
|
15 |
|