Skip to content

GitLab

Unverified Commit 01849b40 authored by Jocelyn's avatar Jocelyn Committed by GitHub
Browse files

add o3-mini support (#2697) 

* add o3-mini support

* fix linter tests
parent a9a0e3ca
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
lm_eval/models/openai_completions.py
View file @ 01849b40
...@@ -291,4 +291,6 @@ class OpenAIChatCompletion(LocalChatCompletion): ...@@ -291,4 +291,6 @@ class OpenAIChatCompletion(LocalChatCompletion):
if "o1" in self.model: if "o1" in self.model:
output.pop("stop") output.pop("stop")
output["temperature"] = 1 output["temperature"] = 1
elif "o3" in self.model:
output.pop("temperature")
return output return output
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment