maulikanalog commited on
Commit
194e86c
1 Parent(s): ed548d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -36,7 +36,7 @@ def solve_problem(problem, max_iterations, reward_threshold, swift_model_id, sag
36
  "api_config": api_configs[ENGINE],
37
  "temperature": float(swift_temperature),
38
  "top_p": float(swift_top_p),
39
- "max_tokens": 2048,
40
  }
41
 
42
  feedback_config = {
@@ -44,7 +44,7 @@ def solve_problem(problem, max_iterations, reward_threshold, swift_model_id, sag
44
  "api_config": api_configs[ENGINE],
45
  "temperature": float(feedback_temperature),
46
  "top_p": float(feedback_top_p),
47
- "max_tokens": 2048,
48
  }
49
 
50
  sage_config = {
@@ -52,7 +52,7 @@ def solve_problem(problem, max_iterations, reward_threshold, swift_model_id, sag
52
  "api_config": api_configs[ENGINE],
53
  "temperature": float(sage_temperature),
54
  "top_p": float(sage_top_p),
55
- "max_tokens": 2048,
56
  }
57
 
58
  # specify the path to the prompt templates
 
36
  "api_config": api_configs[ENGINE],
37
  "temperature": float(swift_temperature),
38
  "top_p": float(swift_top_p),
39
+ "max_tokens": 8192,
40
  }
41
 
42
  feedback_config = {
 
44
  "api_config": api_configs[ENGINE],
45
  "temperature": float(feedback_temperature),
46
  "top_p": float(feedback_top_p),
47
+ "max_tokens": 8192,
48
  }
49
 
50
  sage_config = {
 
52
  "api_config": api_configs[ENGINE],
53
  "temperature": float(sage_temperature),
54
  "top_p": float(sage_top_p),
55
+ "max_tokens": 8192,
56
  }
57
 
58
  # specify the path to the prompt templates