We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 6e498df commit 1bbd562Copy full SHA for 1bbd562
1 file changed
src/tfbench/lm/_google.py
@@ -95,7 +95,7 @@ def _gen(self, prompt: str) -> LMAnswer:
95
config=GenerateContentConfig(
96
system_instruction=[self.instruction],
97
thinking_config=ThinkingConfig(
98
- thinking_budget=EFFORT_TOKEN_MAP[self.effort],
+ thinking_budget=GEMINI_MAP[self.effort],
99
include_thoughts=True,
100
),
101
0 commit comments