kalomaze commited on
Commit
15fec57
·
verified ·
1 Parent(s): 4e2407f

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +78 -0
README.md CHANGED
@@ -8,3 +8,81 @@ base_model: EssentialAI/rnj-1-instruct
8
  WIP derivative checkpoint of [EssentialAI/rnj-1-instruct](https://huggingface.co/EssentialAI/rnj-1-instruct).
9
 
10
  Baseline for tool-calling experiments.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8
  WIP derivative checkpoint of [EssentialAI/rnj-1-instruct](https://huggingface.co/EssentialAI/rnj-1-instruct).
9
 
10
  Baseline for tool-calling experiments.
11
+
12
+ ## Training Config
13
+
14
+ ```toml
15
+ max_steps = 140
16
+ seq_len = 8192
17
+
18
+ [model]
19
+ name = "EssentialAI/rnj-1-instruct"
20
+
21
+ [wandb]
22
+ project = "xml-tool-thinking"
23
+ name = "hex-v6-bs512-rollouts16"
24
+
25
+ [orchestrator.wandb.log_extras]
26
+ samples = true
27
+ interval = 1
28
+
29
+ [orchestrator]
30
+ batch_size = 512
31
+ rollouts_per_example = 16
32
+
33
+ [orchestrator.sampling]
34
+ max_tokens = 512
35
+ temperature = 1.0
36
+
37
+ [[orchestrator.env]]
38
+ id = "hex_encode_xml"
39
+ args = { max_turns = 5, max_chunk = 128, strict_format = true, user_prompt = """Here is a document with semantic XML tags:
40
+
41
+ {doc}
42
+
43
+ Your task: Encode the content of the <{tag_name}> tag to hexadecimal.
44
+
45
+ You must encode in chunks of at most {max_chunk_size} characters at a time.
46
+
47
+ Available tools:
48
+ - get_tag_content: Get the target tag's text and length. No arguments.
49
+ - encode_chunk: Encode a character range to hex. Args: start (int), end (int)
50
+
51
+ <format_rules>
52
+ - Every response must begin with [think]
53
+ - After [/think], include your tool call
54
+ - No text outside of [think]...[/think] and <tool_call>...</tool_call>
55
+ </format_rules>
56
+
57
+ Tool format:
58
+ <tool_call>
59
+ <name>tool_name</name>
60
+ <param name="arg_name">value</param>
61
+ </tool_call>
62
+
63
+ When done, output ONLY the final hex string with no tool calls.
64
+
65
+ Example:
66
+ [think]I need to get the content first.[/think]
67
+ <tool_call>
68
+ <name>get_tag_content</name>
69
+ </tool_call>""" }
70
+
71
+ [trainer.model]
72
+ ac = { freq = 1 }
73
+
74
+ [trainer.optim]
75
+ lr = 1e-6
76
+ max_norm = 0.001
77
+
78
+ [trainer.scheduler]
79
+ type = "linear"
80
+ warmup_steps = 30
81
+ decay_steps = 30
82
+ min_lr = 0
83
+
84
+ [inference.parallel]
85
+ tp = 4
86
+
87
+ [ckpt]
88
+ ```