Skip to content

Commit 1bbd562

Browse files
committed
fix: Gemini use its own EFFORT_TOKEN_MAP
1 parent 6e498df commit 1bbd562

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/tfbench/lm/_google.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,7 @@ def _gen(self, prompt: str) -> LMAnswer:
9595
config=GenerateContentConfig(
9696
system_instruction=[self.instruction],
9797
thinking_config=ThinkingConfig(
98-
thinking_budget=EFFORT_TOKEN_MAP[self.effort],
98+
thinking_budget=GEMINI_MAP[self.effort],
9999
include_thoughts=True,
100100
),
101101
),

0 commit comments

Comments
 (0)