dromerosm commited on
Commit
f93288b
·
verified ·
1 Parent(s): 6db1ddf

Update moa/main.py

Browse files
Files changed (1) hide show
  1. 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': 'llama3-70b-8192'},
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='mixtral-8x7b-32768',
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