gpt4 evaluator: use the model name provided in config

Gapminder · Jan 29, 2024 · 2e7fa8c · 2e7fa8c
1 parent 8b4bbb0
commit 2e7fa8c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/automation-api/yival_experiments/custom_configuration/gpt4_evaluator.py b/automation-api/yival_experiments/custom_configuration/gpt4_evaluator.py
@@ -135,7 +135,7 @@ def evaluate(self, experiment_result: ExperimentResult) -> EvaluatorOutput:
             choices=choices_to_string(self.config.choices)
         )
         response = completion_with_backpff(
-            model="gpt-4",
+            model=self.config.model_name,
             messages=prompt,
             temperature=0.0,
             n=1,