Skip to content

Commit 5191dab

Browse files
committed
use google api
1 parent 7e9b419 commit 5191dab

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

examples/mlx_kernel_optimization/config.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,11 +5,11 @@ log_level: "INFO"
55

66
# LLM configuration
77
llm:
8-
primary_model: "google/gemini-2.0-flash-001"
8+
primary_model: "gemini-2.5-flash-preview-05-20"
99
primary_model_weight: 0.8
10-
secondary_model: "anthropic/claude-sonnet-4"
10+
secondary_model: "gemini-2.5-pro-preview-05-06"
1111
secondary_model_weight: 0.2
12-
api_base: "https://openrouter.ai/api/v1"
12+
api_base: "https://generativelanguage.googleapis.com/v1beta/openai/"
1313
temperature: 0.7
1414
top_p: 0.95
1515
max_tokens: 8192

0 commit comments

Comments
 (0)