test-time-training/.env.example at main · jbarnes850/test-time-training · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
# Environment configuration for kernelbench-rl-env
# Copy this file to .env and fill in your values

# Required - Tinker API for model access
TINKER_API_KEY=your-tinker-api-key-here

# Optional - HuggingFace for dataset access and model uploads
HF_TOKEN=your-huggingface-token-here

# Optional - Weights & Biases for experiment tracking
WANDB_API_KEY=your-wandb-api-key-here

# Optional - Evaluation mode
# KERNELBENCH_EVAL_MODE=full  # 'fast' (5 trials) or 'full' (50 trials)

# Optional - Telemetry output path (defaults to runs/ directory)
# KERNELBENCH_TELEMETRY_PATH=/path/to/telemetry

# Optional - Reward configuration
# KERNELBENCH_REWARD_MODE=normalized
# KERNELBENCH_REWARD_BASELINE_WINDOW=32
# KERNELBENCH_CORRECTNESS_BONUS=0.0