Skip to content
Merged
Show file tree
Hide file tree
Changes from 8 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion configs/basic/alphabet-sort/rl.toml
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ group_size = 8
[orchestrator.train.sampling]
max_completion_tokens = 768

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "alphabet-sort"
env.taskset = { id = "alphabet-sort-v1", min_turns = 3, max_turns = 5, task = { power_per_turn = false } }
env.agent.harness = { id = "null" }
Expand Down
2 changes: 1 addition & 1 deletion configs/basic/hendrycks-sanity/rl.toml
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ batch_size = 128
group_size = 8
seq_len = 8192

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "hendrycks-math"
env.taskset = { id = "math-env-v1", dataset_name = "mikasenghaas/Sanity-Test-R1D-1.5B", dataset_subset = "default" }
env.agent.harness = { id = "null" }
Expand Down
2 changes: 1 addition & 1 deletion configs/basic/reverse-text/rl.toml
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ group_size = 16
[orchestrator.train.sampling]
max_completion_tokens = 128

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand Down
2 changes: 1 addition & 1 deletion configs/basic/wiki-search/rl.toml
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ group_size = 16
[orchestrator.train.sampling]
max_completion_tokens = 512

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "wiki-search"
env.taskset = { id = "wiki-search-v1" }
env.agent.harness = { id = "null" }
Expand Down
2 changes: 1 addition & 1 deletion configs/basic/wordle/rl.toml
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ name = "PrimeIntellect/Qwen3-1.7B-Wordle-SFT"
batch_size = 128
group_size = 8

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "wordle"
env.taskset = { id = "wordle-v1" }
env.player.harness = { id = "null" }
Expand Down
2 changes: 1 addition & 1 deletion configs/ci/integration/alphabet_sort.toml
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ group_size = 8
[orchestrator.train.sampling]
max_completion_tokens = 384

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "alphabet-sort"
env.taskset = { id = "alphabet-sort-v1", min_turns = 2, max_turns = 2, min_names_per_turn = 1, max_names_per_turn = 3, task = { similarity_power = 4, power_per_turn = false } }
env.agent.harness = { id = "null" }
Expand Down
2 changes: 1 addition & 1 deletion configs/ci/integration/reverse-text-lora/resume.toml
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ name = "r8-1e-4"
[orchestrator.train.sampling]
max_completion_tokens = 128

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand Down
2 changes: 1 addition & 1 deletion configs/ci/integration/reverse-text-lora/start.toml
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ name = "r8-1e-4"
[orchestrator.train.sampling]
max_completion_tokens = 128

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand Down
2 changes: 1 addition & 1 deletion configs/ci/integration/reverse-text-moe/start.toml
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ group_size = 16
[orchestrator.train.sampling]
max_completion_tokens = 128

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ lr = 3e-5
[train.sampling]
max_completion_tokens = 128

[[train.env]]
[[train.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand Down
4 changes: 2 additions & 2 deletions configs/ci/integration/reverse-text-rl-opd/start.toml
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ name = "prime-qwen3"
[orchestrator.train.sampling]
max_completion_tokens = 128

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand All @@ -50,7 +50,7 @@ num_examples = 16
[orchestrator.eval.sampling]
max_completion_tokens = 128

[[orchestrator.eval.env]]
[[orchestrator.eval.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand Down
4 changes: 2 additions & 2 deletions configs/ci/integration/reverse-text-rl-sft/start.toml
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@ name = "prime-qwen3"
[orchestrator.train.sampling]
max_completion_tokens = 128

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand All @@ -56,7 +56,7 @@ num_examples = 16
[orchestrator.eval.sampling]
max_completion_tokens = 128

[[orchestrator.eval.env]]
[[orchestrator.eval.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand Down
2 changes: 1 addition & 1 deletion configs/ci/integration/reverse-text/resume.toml
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ group_size = 16
[orchestrator.train.sampling]
max_completion_tokens = 128

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand Down
2 changes: 1 addition & 1 deletion configs/ci/integration/reverse-text/start.toml
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ group_size = 16
[orchestrator.train.sampling]
max_completion_tokens = 128

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand Down
2 changes: 1 addition & 1 deletion configs/ci/nightly-fft/alphabet-sort.toml
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ name = "Qwen/Qwen3-4B-Instruct-2507"
batch_size = 512
group_size = 16

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "alphabet-sort"
env.taskset = { id = "alphabet-sort-v1", min_turns = 3, max_turns = 5, task = { power_per_turn = false } }
env.agent.harness = { id = "null" }
Expand Down
4 changes: 2 additions & 2 deletions configs/ci/nightly-fft/hendrycks-sanity.toml
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ batch_size = 512
group_size = 8
seq_len = 8192

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "hendrycks-math"
env.taskset = { id = "math-env-v1", dataset_name = "mikasenghaas/Sanity-Test-R1D-1.5B", dataset_subset = "default" }
env.agent.harness = { id = "null" }
Expand All @@ -27,7 +27,7 @@ env.agent.runtime = { type = "subprocess" }
[orchestrator.eval]
interval = 50

[[orchestrator.eval.env]]
[[orchestrator.eval.source]]
name = "aime2024"
env.taskset = { id = "aime24-v1" }
env.agent.harness = { id = "null" }
Expand Down
2 changes: 1 addition & 1 deletion configs/ci/nightly-fft/reverse-text.toml
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ name = "PrimeIntellect/Qwen3-0.6B-Reverse-Text-SFT"
batch_size = 128
group_size = 16

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand Down
2 changes: 1 addition & 1 deletion configs/ci/nightly-fft/wiki-search.toml
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ oversampling_factor = 2.0
type = "zero_advantage"
enforce = true

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "wiki-search"
env.taskset = { id = "wiki-search-v1" }
env.agent.harness = { id = "null" }
Expand Down
2 changes: 1 addition & 1 deletion configs/ci/nightly-fft/wordle.toml
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ name = "PrimeIntellect/Qwen3-1.7B-Wordle-SFT"
batch_size = 512
group_size = 16

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "wordle"
env.taskset = { id = "wordle-v1" }
env.player.harness = { id = "null" }
Expand Down
2 changes: 1 addition & 1 deletion configs/ci/nightly/multimodal_color_codeword.toml
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ group_size = 16
[orchestrator.train.sampling]
max_completion_tokens = 64

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "color-codeword"
env.taskset = { id = "color-codeword-v1", images_per_turn = 1 }
env.agent.harness = { id = "null" }
Expand Down
2 changes: 1 addition & 1 deletion configs/debug/algo/echo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ type = "echo"
[orchestrator.algo.roles.user]
alpha = 0.1

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "alphabet-sort"
env.taskset = { id = "alphabet-sort-v1", min_turns = 3, max_turns = 5, task = { power_per_turn = false } }
env.agent.harness = { id = "null" }
Expand Down
4 changes: 2 additions & 2 deletions configs/debug/algo/grpo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ name = "prime-qwen3"
[orchestrator.train.sampling]
max_completion_tokens = 128

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand All @@ -34,7 +34,7 @@ num_examples = 128
[orchestrator.eval.sampling]
max_completion_tokens = 128

[[orchestrator.eval.env]]
[[orchestrator.eval.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand Down
4 changes: 2 additions & 2 deletions configs/debug/algo/max_rl.toml
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ name = "prime-qwen3"
[orchestrator.train.sampling]
max_completion_tokens = 128

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand All @@ -34,7 +34,7 @@ num_examples = 128
[orchestrator.eval.sampling]
max_completion_tokens = 128

[[orchestrator.eval.env]]
[[orchestrator.eval.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand Down
10 changes: 5 additions & 5 deletions configs/debug/algo/mixed_grpo_opd.toml
Original file line number Diff line number Diff line change
Expand Up @@ -31,22 +31,22 @@ name = "prime-qwen3"
[orchestrator.train.sampling]
max_completion_tokens = 128

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "reverse-text-grpo"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
env.agent.runtime = { type = "subprocess" }

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "reverse-text-opd"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
env.agent.runtime = { type = "subprocess" }

[orchestrator.train.env.algo]
[orchestrator.train.source.algo]
type = "opd"

[orchestrator.train.env.algo.teacher]
[orchestrator.train.source.algo.teacher]
name = "PrimeIntellect/Qwen3-0.6B-Reverse-Text-RL"
base_url = ["http://localhost:8001/v1"]

Expand All @@ -57,7 +57,7 @@ num_examples = 128
[orchestrator.eval.sampling]
max_completion_tokens = 128

[[orchestrator.eval.env]]
[[orchestrator.eval.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand Down
4 changes: 2 additions & 2 deletions configs/debug/algo/opd.toml
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ name = "prime-qwen3"
[orchestrator.train.sampling]
max_completion_tokens = 128

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand All @@ -46,7 +46,7 @@ num_examples = 128
[orchestrator.eval.sampling]
max_completion_tokens = 128

[[orchestrator.eval.env]]
[[orchestrator.eval.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand Down
4 changes: 2 additions & 2 deletions configs/debug/algo/opd_lora.toml
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ name = "prime-qwen3"
[orchestrator.train.sampling]
max_completion_tokens = 128

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand All @@ -46,7 +46,7 @@ num_examples = 128
[orchestrator.eval.sampling]
max_completion_tokens = 128

[[orchestrator.eval.env]]
[[orchestrator.eval.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand Down
4 changes: 2 additions & 2 deletions configs/debug/algo/rae.toml
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ name = "prime-qwen3"
[orchestrator.train.sampling]
max_completion_tokens = 512

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "kuhn-poker"
env.taskset = { id = "kuhn-poker-v1" }
env.player0.harness = { id = "null" }
Expand All @@ -36,7 +36,7 @@ num_examples = 16
[orchestrator.eval.sampling]
max_completion_tokens = 512

[[orchestrator.eval.env]]
[[orchestrator.eval.source]]
name = "kuhn-poker"
env.taskset = { id = "kuhn-poker-v1" }
env.player0.harness = { id = "null" }
Expand Down
4 changes: 2 additions & 2 deletions configs/debug/algo/self_distill.toml
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ name = "prime-qwen3"
[orchestrator.train.sampling]
max_completion_tokens = 128

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand All @@ -42,7 +42,7 @@ num_examples = 128
[orchestrator.eval.sampling]
max_completion_tokens = 128

[[orchestrator.eval.env]]
[[orchestrator.eval.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand Down
4 changes: 2 additions & 2 deletions configs/debug/algo/sft_distill.toml
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ name = "prime-qwen3"
[orchestrator.train.sampling]
max_completion_tokens = 128

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand All @@ -51,7 +51,7 @@ num_examples = 128
[orchestrator.eval.sampling]
max_completion_tokens = 128

[[orchestrator.eval.env]]
[[orchestrator.eval.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand Down
4 changes: 2 additions & 2 deletions configs/debug/algo/sft_distill_lora.toml
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ name = "prime-qwen3"
[orchestrator.train.sampling]
max_completion_tokens = 128

[[orchestrator.train.env]]
[[orchestrator.train.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand All @@ -51,7 +51,7 @@ num_examples = 128
[orchestrator.eval.sampling]
max_completion_tokens = 128

[[orchestrator.eval.env]]
[[orchestrator.eval.source]]
name = "reverse-text"
env.taskset = { id = "reverse-text-v1" }
env.agent.harness = { id = "null" }
Expand Down
Loading
Loading