DrishtiSharma commited on
Commit
9e0238f
Β·
verified Β·
1 Parent(s): 6c54090

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -16,7 +16,7 @@ default_config = {
16
  layer_agent_config_def = {
17
  "layer_agent_1": {
18
  "system_prompt": "Think through your response step by step. {helper_response}",
19
- "model_name": "llama3-8b-8192"
20
  },
21
  "layer_agent_2": {
22
  "system_prompt": "Respond with a thought and then your response to the question. {helper_response}",
@@ -25,7 +25,7 @@ layer_agent_config_def = {
25
  },
26
  "layer_agent_3": {
27
  "system_prompt": "You are an expert at logic and reasoning. Always take a logical approach to the answer. {helper_response}",
28
- "model_name": "llama3-8b-8192"
29
  },
30
 
31
  }
@@ -33,7 +33,7 @@ layer_agent_config_def = {
33
  # Recommended Configuration
34
 
35
  rec_config = {
36
- "main_model": "llama3-70b-8192",
37
  "cycles": 2,
38
  "layer_agent_config": {}
39
  }
@@ -41,17 +41,17 @@ rec_config = {
41
  layer_agent_config_rec = {
42
  "layer_agent_1": {
43
  "system_prompt": "Think through your response step by step. {helper_response}",
44
- "model_name": "llama3-8b-8192",
45
  "temperature": 0.1
46
  },
47
  "layer_agent_2": {
48
  "system_prompt": "Respond with a thought and then your response to the question. {helper_response}",
49
- "model_name": "llama3-8b-8192",
50
  "temperature": 0.2
51
  },
52
  "layer_agent_3": {
53
  "system_prompt": "You are an expert at logic and reasoning. Always take a logical approach to the answer. {helper_response}",
54
- "model_name": "llama3-8b-8192",
55
  "temperature": 0.4
56
  },
57
  "layer_agent_4": {
 
16
  layer_agent_config_def = {
17
  "layer_agent_1": {
18
  "system_prompt": "Think through your response step by step. {helper_response}",
19
+ "model_name": "llama-3.3-70b-versatile"
20
  },
21
  "layer_agent_2": {
22
  "system_prompt": "Respond with a thought and then your response to the question. {helper_response}",
 
25
  },
26
  "layer_agent_3": {
27
  "system_prompt": "You are an expert at logic and reasoning. Always take a logical approach to the answer. {helper_response}",
28
+ "model_name": "llama-3.3-70b-versatile"
29
  },
30
 
31
  }
 
33
  # Recommended Configuration
34
 
35
  rec_config = {
36
+ "main_model": "llama-3.3-70b-versatile",
37
  "cycles": 2,
38
  "layer_agent_config": {}
39
  }
 
41
  layer_agent_config_rec = {
42
  "layer_agent_1": {
43
  "system_prompt": "Think through your response step by step. {helper_response}",
44
+ "model_name": "llama-3.3-70b-versatile",
45
  "temperature": 0.1
46
  },
47
  "layer_agent_2": {
48
  "system_prompt": "Respond with a thought and then your response to the question. {helper_response}",
49
+ "model_name": "llama-3.3-70b-versatile",
50
  "temperature": 0.2
51
  },
52
  "layer_agent_3": {
53
  "system_prompt": "You are an expert at logic and reasoning. Always take a logical approach to the answer. {helper_response}",
54
+ "model_name": "llama-3.3-70b-versatile",
55
  "temperature": 0.4
56
  },
57
  "layer_agent_4": {