BtB-ExpC commited on
Commit
aa2d099
·
1 Parent(s): f7e044d

gpt41-mini-zero ModelNames keys aligned again

Browse files
Files changed (1) hide show
  1. config/chain_configs.py +4 -4
config/chain_configs.py CHANGED
@@ -55,7 +55,7 @@ chain_configs = {
55
  "distractors": {
56
  "class": DistractorsChain,
57
  "template_standardize": template_standardize_exercise,
58
- "llm_standardize": llms[ModelNames.GPT4_MINI_ZERO], # Always fixed
59
  "template_distractors_brainstorm_1": template_distractors_brainstorm_1,
60
  "template_distractors_brainstorm_2": template_distractors_brainstorm_2,
61
  "llm_brainstorm_1": llms[ModelNames.GPT4_LOW],
@@ -66,13 +66,13 @@ chain_configs = {
66
  "learning_objectives": {
67
  "class": LearningObjectivesChain,
68
  "template_standardize": template_standardize_studytext,
69
- "llm_standardize": llms[ModelNames.GPT4_MINI_ZERO], # Always fixed
70
  "template_gen_prompt_a": template_gen_prompt_a,
71
  "template_gen_prompt_b": template_gen_prompt_b,
72
  "default_llm_a": llms[ModelNames.O1_HIGH_REASONING],
73
  "default_llm_b": llms[ModelNames.O3_MINI_HIGH_REASONING],
74
  "template_sanitize": template_sanitize_learning_objectives,
75
- "llm_sanitize": llms[ModelNames.GPT4_MINI_ZERO],
76
  },
77
  "fluster": {
78
  "class": FlusterWritingChain,
@@ -84,7 +84,7 @@ chain_configs = {
84
  "template_refine_fluster": template_refine_fluster,
85
  "llm_refine": llms[ModelNames.GPT4_ZERO],
86
  "template_sanitize": template_sanitize_fluster,
87
- "llm_sanitize": llms[ModelNames.GPT4_MINI_ZERO],
88
  "template_structurize": template_isolate_exercises,
89
  "llm_structurize": llms[ModelNames.GPT4_ZERO],
90
  "template_fix_exercise": template_fix_exercise,
 
55
  "distractors": {
56
  "class": DistractorsChain,
57
  "template_standardize": template_standardize_exercise,
58
+ "llm_standardize": llms[ModelNames.GPT41_MINI_ZERO], # Always fixed
59
  "template_distractors_brainstorm_1": template_distractors_brainstorm_1,
60
  "template_distractors_brainstorm_2": template_distractors_brainstorm_2,
61
  "llm_brainstorm_1": llms[ModelNames.GPT4_LOW],
 
66
  "learning_objectives": {
67
  "class": LearningObjectivesChain,
68
  "template_standardize": template_standardize_studytext,
69
+ "llm_standardize": llms[ModelNames.GPT41_MINI_ZERO], # Always fixed
70
  "template_gen_prompt_a": template_gen_prompt_a,
71
  "template_gen_prompt_b": template_gen_prompt_b,
72
  "default_llm_a": llms[ModelNames.O1_HIGH_REASONING],
73
  "default_llm_b": llms[ModelNames.O3_MINI_HIGH_REASONING],
74
  "template_sanitize": template_sanitize_learning_objectives,
75
+ "llm_sanitize": llms[ModelNames.GPT41_MINI_ZERO],
76
  },
77
  "fluster": {
78
  "class": FlusterWritingChain,
 
84
  "template_refine_fluster": template_refine_fluster,
85
  "llm_refine": llms[ModelNames.GPT4_ZERO],
86
  "template_sanitize": template_sanitize_fluster,
87
+ "llm_sanitize": llms[ModelNames.GPT41_MINI_ZERO],
88
  "template_structurize": template_isolate_exercises,
89
  "llm_structurize": llms[ModelNames.GPT4_ZERO],
90
  "template_fix_exercise": template_fix_exercise,