Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+tokenizer.json filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,76 @@
+---
+base_model: google/gemma-4-E2B-it-qat-mobile-transformers
+library_name: peft
+license: gemma
+pipeline_tag: text-generation
+language: [en]
+tags:
+- lora
+- peft
+- strands-agents
+- code
+- gemma-4
+- domain-adaptation
+---
+
+# strands-gemma4-e2b — Strands Agents expert (Gemma 4 E2B LoRA)
+
+LoRA adapter that post-tunes **google/gemma-4-E2B-it-qat-mobile-transformers** on the
+full **Strands Agents / Strands Robots** codebase + docs corpus, turning the mobile-class
+Gemma 4 E2B into a Strands-Agents domain expert.
+
+Sister models: [cagataydev/strands-qwen3-vl-2b](https://huggingface.co/cagataydev/strands-qwen3-vl-2b)
+(same corpus, Qwen3-VL-2B backbone). Part of the self-learning-model research project
+([github.com/cagataycali/slm](https://github.com/cagataycali/slm)).
+
+## Results
+
+| Metric | Before | After | Δ |
+|---|---|---|---|
+| Holdout NLL (strands corpus) | **2.689** | **1.260** | **−1.43 (−53%)** |
+| Train loss (600 steps) | 3.07 | 0.96 | |
+
+## Training
+
+- **Corpus**: 2,288 documents (~5M tokens) — strands-agents SDK source, strands-tools,
+  strands-robots, docs, examples, and Golden-200 Q&A pairs
+- **Method**: QAT checkpoint dequantized to bf16, then LoRA on all language-model
+  attention + MLP projections (`q,k,v,o,gate,up,down_proj`), vision tower untouched
+- **Config**: r=32, alpha=64, dropout=0.05, 600 steps, bs 2 × accum 4, lr 1e-4 cosine,
+  block 1024, AdamW, gradient checkpointing
+- **Trainable**: 48.3M params (adapter only; base frozen)
+- **Hardware**: 1× NVIDIA L40S, ~2,570 tok/s
+
+## Usage
+
+```python
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer
+from peft import PeftModel
+
+BASE = "google/gemma-4-E2B-it-qat-mobile-transformers"
+tok = AutoTokenizer.from_pretrained(BASE)
+model = AutoModelForCausalLM.from_pretrained(BASE, dtype=torch.bfloat16, device_map="auto")
+model = PeftModel.from_pretrained(model, "cagataydev/strands-gemma4-e2b")
+
+msgs = [{"role": "user", "content": "How do I create a custom tool in Strands Agents?"}]
+ids = tok.apply_chat_template(msgs, add_generation_prompt=True, return_tensors="pt").to(model.device)
+out = model.generate(ids, max_new_tokens=256)
+print(tok.decode(out[0, ids.shape[1]:], skip_special_tokens=True))
+```
+
+> Note: the base is a QAT (quantization-aware-training) checkpoint. Training was done on
+> the dequantized bf16 weights; for training-compatible loading, dequantize QAT wrappers
+> or load in bf16 as above.
+
+## Intended use & limitations
+
+- Domain expert for the Strands Agents ecosystem (SDK APIs, tools, patterns, robots).
+- Inherits Gemma 4 license/usage terms. Not evaluated for general-purpose safety beyond base.
+- Trained on a code corpus snapshot (July 2026); APIs may drift.
+
+## Reproducibility
+
+Training script: `strands_tune/train_lora_any.py` in the research repo
+(`--dequant-qat --targets attn_mlp --steps 600 --bs 2 --accum 4 --lr 1e-4 --r 32`).
+Full step log in `train_log.json` in this repo.

adapter_config.json

@@ -0,0 +1,38 @@
+{
+  "alora_invocation_tokens": null,
+  "alpha_pattern": {},
+  "arrow_config": null,
+  "auto_mapping": null,
+  "base_model_name_or_path": "google/gemma-4-E2B-it-qat-mobile-transformers",
+  "bias": "none",
+  "corda_config": null,
+  "ensure_weight_tying": false,
+  "eva_config": null,
+  "exclude_modules": null,
+  "fan_in_fan_out": false,
+  "inference_mode": true,
+  "init_lora_weights": true,
+  "layer_replication": null,
+  "layers_pattern": null,
+  "layers_to_transform": null,
+  "loftq_config": {},
+  "lora_alpha": 64,
+  "lora_bias": false,
+  "lora_dropout": 0.05,
+  "megatron_config": null,
+  "megatron_core": "megatron.core",
+  "modules_to_save": null,
+  "peft_type": "LORA",
+  "peft_version": "0.18.1",
+  "qalora_group_size": 16,
+  "r": 32,
+  "rank_pattern": {},
+  "revision": null,
+  "target_modules": ".*language_model.*(q_proj|k_proj|v_proj|o_proj|gate_proj|up_proj|down_proj)$",
+  "target_parameters": null,
+  "task_type": "CAUSAL_LM",
+  "trainable_token_indices": null,
+  "use_dora": false,
+  "use_qalora": false,
+  "use_rslora": false
+}

adapter_model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c66496a2c0f91df635a14f703cf25ceae5e58e3392e2b1a2f1b2b03b46ed5b1c
+size 193326648

chat_template.jinja

@@ -0,0 +1,360 @@
+{%- macro format_parameters(properties, required, filter_keys=false) -%}
+    {%- set standard_keys = ['description', 'type', 'properties', 'required', 'nullable'] -%}
+    {%- set ns = namespace(found_first=false) -%}
+    {%- for key, value in properties | dictsort -%}
+        {%- set add_comma = false -%}
+        {%- if not filter_keys or key not in standard_keys -%}
+            {%- if ns.found_first %},{% endif -%}
+            {%- set ns.found_first = true -%}
+            {{ key }}:{
+            {%- if value['description'] -%}
+                description:<|"|>{{ value['description'] }}<|"|>
+                {%- set add_comma = true -%}
+            {%- endif -%}
+            {%- if value['type'] | upper == 'STRING' -%}
+                {%- if value['enum'] -%}
+                    {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
+                    enum:{{ format_argument(value['enum']) }}
+                {%- endif -%}
+            {%- elif value['type'] | upper == 'ARRAY' -%}
+                {%- if value['items'] is mapping and value['items'] -%}
+                    {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
+                    items:{
+                    {%- set ns_items = namespace(found_first=false) -%}
+                    {%- for item_key, item_value in value['items'] | dictsort -%}
+                        {%- if item_value is not none -%}
+                            {%- if ns_items.found_first %},{% endif -%}
+                            {%- set ns_items.found_first = true -%}
+                            {%- if item_key == 'properties' -%}
+                                properties:{
+                                {%- if item_value is mapping -%}
+                                    {{- format_parameters(item_value, value['items']['required'] | default([])) -}}
+                                {%- endif -%}
+                                }
+                            {%- elif item_key == 'required' -%}
+                                required:[
+                                {%- for req_item in item_value -%}
+                                    <|"|>{{- req_item -}}<|"|>
+                                    {%- if not loop.last %},{% endif -%}
+                                {%- endfor -%}
+                                ]
+                            {%- elif item_key == 'type' -%}
+                                {%- if item_value is string -%}
+                                    type:{{ format_argument(item_value | upper) }}
+                                {%- else -%}
+                                    type:{{ format_argument(item_value | map('upper') | list) }}
+                                {%- endif -%}
+                            {%- else -%}
+                                {{ item_key }}:{{ format_argument(item_value) }}
+                            {%- endif -%}
+                        {%- endif -%}
+                    {%- endfor -%}
+                    }
+                {%- endif -%}
+            {%- endif -%}
+            {%- if value['nullable'] %}
+                {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
+                nullable:true
+            {%- endif -%}
+            {%- if value['type'] | upper == 'OBJECT' -%}
+                {%- if value['properties'] is defined and value['properties'] is mapping -%}
+                    {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
+                    properties:{
+                    {{- format_parameters(value['properties'], value['required'] | default([])) -}}
+                    }
+                {%- elif value is mapping -%}
+                    {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
+                    properties:{
+                    {{- format_parameters(value, value['required'] | default([]), filter_keys=true) -}}
+                    }
+                {%- endif -%}
+                {%- if value['required'] -%}
+                    {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
+                    required:[
+                    {%- for item in value['required'] | default([]) -%}
+                        <|"|>{{- item -}}<|"|>
+                        {%- if not loop.last %},{% endif -%}
+                    {%- endfor -%}
+                    ]
+                {%- endif -%}
+            {%- endif -%}
+            {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
+            type:<|"|>{{ value['type'] | upper }}<|"|>}
+        {%- endif -%}
+    {%- endfor -%}
+{%- endmacro -%}
+{%- macro format_function_declaration(tool_data) -%}
+    declaration:{{- tool_data['function']['name'] -}}{description:<|"|>{{- tool_data['function']['description'] -}}<|"|>
+    {%- set params = tool_data['function']['parameters'] -%}
+    {%- if params -%}
+        ,parameters:{
+        {%- if params['properties'] -%}
+            properties:{ {{- format_parameters(params['properties'], params['required']) -}} },
+        {%- endif -%}
+        {%- if params['required'] -%}
+            required:[
+            {%- for item in params['required'] -%}
+                <|"|>{{- item -}}<|"|>
+                {{- ',' if not loop.last -}}
+            {%- endfor -%}
+            ],
+        {%- endif -%}
+        {%- if params['type'] -%}
+            type:<|"|>{{- params['type'] | upper -}}<|"|>}
+        {%- endif -%}
+    {%- endif -%}
+    {%- if 'response' in tool_data['function'] -%}
+        {%- set response_declaration = tool_data['function']['response'] -%}
+        ,response:{
+        {%- if response_declaration['description'] -%}
+            description:<|"|>{{- response_declaration['description'] -}}<|"|>,
+        {%- endif -%}
+        {%- if response_declaration['type'] | upper == 'OBJECT' -%}
+            type:<|"|>{{- response_declaration['type'] | upper -}}<|"|>}
+        {%- endif -%}
+    {%- endif -%}
+    }
+{%- endmacro -%}
+{%- macro format_argument(argument, escape_keys=True) -%}
+    {%- if argument is string -%}
+        {{- '<|"|>' + argument + '<|"|>' -}}
+    {%- elif argument is boolean -%}
+        {{- 'true' if argument else 'false' -}}
+    {%- elif argument is mapping -%}
+        {{- '{' -}}
+        {%- set ns = namespace(found_first=false) -%}
+        {%- for key, value in argument | dictsort -%}
+            {%- if ns.found_first %},{% endif -%}
+            {%- set ns.found_first = true -%}
+            {%- if escape_keys -%}
+                {{- '<|"|>' + key + '<|"|>' -}}
+            {%- else -%}
+                {{- key -}}
+            {%- endif -%}
+            :{{- format_argument(value, escape_keys=escape_keys) -}}
+        {%- endfor -%}
+        {{- '}' -}}
+    {%- elif argument is sequence -%}
+        {{- '[' -}}
+        {%- for item in argument -%}
+            {{- format_argument(item, escape_keys=escape_keys) -}}
+            {%- if not loop.last %},{% endif -%}
+        {%- endfor -%}
+        {{- ']' -}}
+    {%- else -%}
+        {{- argument -}}
+    {%- endif -%}
+{%- endmacro -%}
+{%- macro strip_thinking(text) -%}
+    {%- set ns = namespace(result='') -%}
+    {%- for part in text.split('<channel|>') -%}
+        {%- if '<|channel>' in part -%}
+            {%- set ns.result = ns.result + part.split('<|channel>')[0] -%}
+        {%- else -%}
+            {%- set ns.result = ns.result + part -%}
+        {%- endif -%}
+    {%- endfor -%}
+    {{- ns.result | trim -}}
+{%- endmacro -%}
+
+{%- macro format_tool_response_block(tool_name, response) -%}
+    {{- '<|tool_response>' -}}
+    {%- if response is mapping -%}
+        {{- 'response:' + tool_name + '{' -}}
+        {%- for key, value in response | dictsort -%}
+            {{- key -}}:{{- format_argument(value, escape_keys=False) -}}
+            {%- if not loop.last %},{% endif -%}
+        {%- endfor -%}
+        {{- '}' -}}
+    {%- else -%}
+        {{- 'response:' + tool_name + '{value:' + format_argument(response, escape_keys=False) + '}' -}}
+    {%- endif -%}
+    {{- '<tool_response|>' -}}
+{%- endmacro -%}
+
+{%- set ns = namespace(prev_message_type=None) -%}
+{%- set loop_messages = messages -%}
+{{- bos_token -}}
+{#- Handle System/Tool Definitions Block -#}
+{%- if (enable_thinking is defined and enable_thinking) or tools or messages[0]['role'] in ['system', 'developer'] -%}
+    {{- '<|turn>system\n' -}}
+    {#- Inject Thinking token at the very top of the FIRST system turn -#}
+    {%- if enable_thinking is defined and enable_thinking -%}
+        {{- '<|think|>\n' -}}
+        {%- set ns.prev_message_type = 'think' -%}
+    {%- endif -%}
+    {%- if messages[0]['role'] in ['system', 'developer'] -%}
+        {%- if messages[0]['content'] is string -%}
+            {{- messages[0]['content'] | trim -}}
+        {%- elif messages[0]['content'] is sequence -%}
+            {%- for item in messages[0]['content'] -%}
+                {{- item['text'] | trim + ' '-}}
+            {%- endfor -%}
+        {%- endif -%}
+        {%- set loop_messages = messages[1:] -%}
+    {%- endif -%}
+    {%- if tools -%}
+        {%- for tool in tools %}
+            {{- '<|tool>' -}}
+            {{- format_function_declaration(tool) | trim -}}
+            {{- '<tool|>' -}}
+        {%- endfor %}
+        {%- set ns.prev_message_type = 'tool' -%}
+    {%- endif -%}
+    {{- '<turn|>\n' -}}
+{%- endif %}
+
+{#- Pre-scan: find last user message index for reasoning guard -#}
+{%- set ns_turn = namespace(last_user_idx=-1) -%}
+{%- for i in range(loop_messages | length) -%}
+    {%- if loop_messages[i]['role'] == 'user' -%}
+        {%- set ns_turn.last_user_idx = i -%}
+    {%- endif -%}
+{%- endfor -%}
+
+{#- Loop through messages -#}
+{%- for message in loop_messages -%}
+    {%- if message['role'] != 'tool' -%}
+    {%- set ns.prev_message_type = None -%}
+    {%- set role = 'model' if message['role'] == 'assistant' else message['role'] -%}
+    {#- Detect continuation: suppress duplicate <|turn>model when previous non-tool message was also assistant -#}
+    {%- set prev_nt = namespace(role=None, found=false) -%}
+    {%- if loop.index0 > 0 -%}
+        {%- for j in range(loop.index0 - 1, -1, -1) -%}
+            {%- if not prev_nt.found -%}
+                {%- if loop_messages[j]['role'] != 'tool' -%}
+                    {%- set prev_nt.role = loop_messages[j]['role'] -%}
+                    {%- set prev_nt.found = true -%}
+                {%- endif -%}
+            {%- endif -%}
+        {%- endfor -%}
+    {%- endif -%}
+    {%- set continue_same_model_turn = (role == 'model' and prev_nt.role == 'assistant') -%}
+    {%- if not continue_same_model_turn -%}
+        {{- '<|turn>' + role + '\n' }}
+    {%- endif -%}
+
+    {#- Render reasoning/reasoning_content as thinking channel -#}
+    {%- set thinking_text = message.get('reasoning') or message.get('reasoning_content') -%}
+    {%- if thinking_text and loop.index0 > ns_turn.last_user_idx and message.get('tool_calls') -%}
+        {{- '<|channel>thought\n' + thinking_text + '\n<channel|>' -}}
+    {%- endif -%}
+
+            {%- if message['tool_calls'] -%}
+                {%- for tool_call in message['tool_calls'] -%}
+                    {%- set function = tool_call['function'] -%}
+                    {{- '<|tool_call>call:' + function['name'] + '{' -}}
+                    {%- if function['arguments'] is mapping -%}
+                        {%- set ns_args = namespace(found_first=false) -%}
+                        {%- for key, value in function['arguments'] | dictsort -%}
+                            {%- if ns_args.found_first %},{% endif -%}
+                            {%- set ns_args.found_first = true -%}
+                            {{- key -}}:{{- format_argument(value, escape_keys=False) -}}
+                        {%- endfor -%}
+                    {%- elif function['arguments'] is string -%}
+                        {{- function['arguments'] -}}
+                    {%- endif -%}
+                    {{- '}<tool_call|>' -}}
+                {%- endfor -%}
+                {%- set ns.prev_message_type = 'tool_call' -%}
+            {%- endif -%}
+
+            {%- set ns_tr_out = namespace(flag=false) -%}
+            {%- if message.get('tool_responses') -%}
+                {#- Legacy: tool_responses embedded on the assistant message (Google/Gemma native) -#}
+                {%- for tool_response in message['tool_responses'] -%}
+                    {{- format_tool_response_block(tool_response['name'] | default('unknown'), tool_response['response']) -}}
+                    {%- set ns_tr_out.flag = true -%}
+                    {%- set ns.prev_message_type = 'tool_response' -%}
+                {%- endfor -%}
+            {%- elif message.get('tool_calls') -%}
+                {#- OpenAI Chat Completions: forward-scan consecutive role:tool messages -#}
+                {%- set ns_tool_scan = namespace(stopped=false) -%}
+                {%- for k in range(loop.index0 + 1, loop_messages | length) -%}
+                    {%- if ns_tool_scan.stopped -%}
+                    {%- elif loop_messages[k]['role'] != 'tool' -%}
+                        {%- set ns_tool_scan.stopped = true -%}
+                    {%- else -%}
+                        {%- set follow = loop_messages[k] -%}
+                        {#- Resolve tool_call_id to function name -#}
+                        {%- set ns_tname = namespace(name=follow.get('name') | default('unknown')) -%}
+                        {%- for tc in message['tool_calls'] -%}
+                            {%- if tc.get('id') == follow.get('tool_call_id') -%}
+                                {%- set ns_tname.name = tc['function']['name'] -%}
+                            {%- endif -%}
+                        {%- endfor -%}
+                        {#- Handle content as string or content-parts array -#}
+                        {%- set tool_body = follow.get('content') -%}
+                        {%- if tool_body is string -%}
+                            {{- format_tool_response_block(ns_tname.name, tool_body) -}}
+                        {%- elif tool_body is sequence and tool_body is not string -%}
+                            {%- set ns_txt = namespace(s='') -%}
+                            {%- for part in tool_body -%}
+                                {%- if part.get('type') == 'text' -%}
+                                    {%- set ns_txt.s = ns_txt.s + (part.get('text') | default('')) -%}
+                                {%- endif -%}
+                            {%- endfor -%}
+                            {{- format_tool_response_block(ns_tname.name, ns_txt.s) -}}
+                            {%- for part in tool_body -%}
+                                {%- if part.get('type') == 'image' -%}
+                                    {{- '<|image|>' -}}
+                                {%- elif part.get('type') == 'audio' -%}
+                                    {{- '<|audio|>' -}}
+                                {%- elif part.get('type') == 'video' -%}
+                                    {{- '<|video|>' -}}
+                                {%- endif -%}
+                            {%- endfor -%}
+                        {%- else -%}
+                            {{- format_tool_response_block(ns_tname.name, tool_body) -}}
+                        {%- endif -%}
+                        {%- set ns_tr_out.flag = true -%}
+                        {%- set ns.prev_message_type = 'tool_response' -%}
+                    {%- endif -%}
+                {%- endfor -%}
+            {%- endif -%}
+
+            {%- set captured_content -%}
+            {%- if message['content'] is string -%}
+                {%- if role == 'model' -%}
+                    {{- strip_thinking(message['content']) -}}
+                {%- else -%}
+                    {{- message['content'] | trim -}}
+                {%- endif -%}
+            {%- elif message['content'] is sequence -%}
+                {%- for item in message['content'] -%}
+                    {%- if item['type'] == 'text' -%}
+                        {%- if role == 'model' -%}
+                            {{- strip_thinking(item['text']) -}}
+                        {%- else -%}
+                            {{- item['text'] | trim -}}
+                        {%- endif -%}
+                    {%- elif item['type'] == 'image' -%}
+                        {{- '<|image|>' -}}
+                        {%- set ns.prev_message_type = 'image' -%}
+                    {%- elif item['type'] == 'audio' -%}
+                        {{- '<|audio|>' -}}
+                        {%- set ns.prev_message_type = 'audio' -%}
+                    {%- elif item['type'] == 'video' -%}
+                        {{- '<|video|>' -}}
+                        {%- set ns.prev_message_type = 'video' -%}
+                    {%- endif -%}
+                {%- endfor -%}
+            {%- endif -%}
+            {%- endset -%}
+
+            {{- captured_content -}}
+            {%- set has_content = captured_content | trim | length > 0 -%}
+
+        {%- if ns.prev_message_type == 'tool_call' and not ns_tr_out.flag -%}
+            {{- '<|tool_response>' -}}
+        {%- elif not (ns_tr_out.flag and not has_content) -%}
+            {{- '<turn|>\n' -}}
+        {%- endif -%}
+    {%- endif -%}
+{%- endfor -%}
+
+{%- if add_generation_prompt -%}
+    {%- if ns.prev_message_type != 'tool_response' and ns.prev_message_type != 'tool_call' -%}
+        {{- '<|turn>model\n' -}}
+    {%- endif -%}
+{%- endif -%}

tokenizer.json

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:55e53b7789842253453fe481c7b5c4ce98cd61654cb333bb39a78e9ed824871d
+size 32169725

tokenizer_config.json

@@ -0,0 +1,96 @@
+{
+  "audio_token": "<|audio|>",
+  "backend": "tokenizers",
+  "boa_token": "<|audio>",
+  "boi_token": "<|image>",
+  "bos_token": "<bos>",
+  "eoa_token": "<audio|>",
+  "eoc_token": "<channel|>",
+  "eoi_token": "<image|>",
+  "eos_token": "<eos>",
+  "eot_token": "<turn|>",
+  "escape_token": "<|\"|>",
+  "etc_token": "<tool_call|>",
+  "etd_token": "<tool|>",
+  "etr_token": "<tool_response|>",
+  "extra_special_tokens": [
+    "<|video|>"
+  ],
+  "image_token": "<|image|>",
+  "is_local": false,
+  "local_files_only": false,
+  "mask_token": "<mask>",
+  "model_max_length": 1000000000000000019884624838656,
+  "model_specific_special_tokens": {
+    "audio_token": "<|audio|>",
+    "boa_token": "<|audio>",
+    "boi_token": "<|image>",
+    "eoa_token": "<audio|>",
+    "eoc_token": "<channel|>",
+    "eoi_token": "<image|>",
+    "eot_token": "<turn|>",
+    "escape_token": "<|\"|>",
+    "etc_token": "<tool_call|>",
+    "etd_token": "<tool|>",
+    "etr_token": "<tool_response|>",
+    "image_token": "<|image|>",
+    "soc_token": "<|channel>",
+    "sot_token": "<|turn>",
+    "stc_token": "<|tool_call>",
+    "std_token": "<|tool>",
+    "str_token": "<|tool_response>",
+    "think_token": "<|think|>"
+  },
+  "pad_token": "<pad>",
+  "padding_side": "left",
+  "processor_class": "Gemma4Processor",
+  "response_schema": {
+    "properties": {
+      "content": {
+        "type": "string"
+      },
+      "role": {
+        "const": "assistant"
+      },
+      "thinking": {
+        "type": "string"
+      },
+      "tool_calls": {
+        "items": {
+          "properties": {
+            "function": {
+              "properties": {
+                "arguments": {
+                  "additionalProperties": {},
+                  "type": "object",
+                  "x-parser": "gemma4-tool-call"
+                },
+                "name": {
+                  "type": "string"
+                }
+              },
+              "type": "object",
+              "x-regex": "call\\:(?P<name>\\w+)(?P<arguments>\\{.*\\})"
+            },
+            "type": {
+              "const": "function"
+            }
+          },
+          "type": "object"
+        },
+        "type": "array",
+        "x-regex-iterator": "<\\|tool_call>(.*?)<tool_call\\|>"
+      }
+    },
+    "type": "object",
+    "x-regex": "(\\<\\|channel\\>thought\\n(?P<thinking>.*?)\\<channel\\|\\>)?(?P<tool_calls>\\<\\|tool_call\\>.*\\<tool_call\\|\\>)?(?P<content>(?:(?!\\<turn\\|\\>)(?!\\<\\|tool_response\\>).)+)?(?:\\<turn\\|\\>|\\<\\|tool_response\\>)?"
+  },
+  "soc_token": "<|channel>",
+  "sot_token": "<|turn>",
+  "stc_token": "<|tool_call>",
+  "std_token": "<|tool>",
+  "str_token": "<|tool_response>",
+  "think_token": "<|think|>",
+  "tokenizer_class": "GemmaTokenizer",
+  "unk_token": "<unk>"
+}

train_log.json

@@ -0,0 +1,387 @@
+{
+  "model": "google/gemma-4-E2B-it-qat-mobile-transformers",
+  "args": {
+    "model": "google/gemma-4-E2B-it-qat-mobile-transformers",
+    "steps": 600,
+    "bs": 2,
+    "accum": 4,
+    "lr": 0.0001,
+    "r": 32,
+    "block": 1024,
+    "load_4bit": false,
+    "dequant_qat": true,
+    "targets": "attn_mlp",
+    "corpus": "/home/ubuntu/self-learning-model/strands_tune/corpus.jsonl",
+    "out": "artifacts/strands_gemma4_e2b_lora"
+  },
+  "nll_before": 2.68935369849205,
+  "nll_after": 1.2602788805961609,
+  "log": [
+    {
+      "step": 1,
+      "loss": 3.0693,
+      "lr": 9.999931461237134e-05,
+      "tok_s": 2557
+    },
+    {
+      "step": 10,
+      "loss": 1.7782,
+      "lr": 9.993147673772868e-05,
+      "tok_s": 2594
+    },
+    {
+      "step": 20,
+      "loss": 1.5848,
+      "lr": 9.972609476841367e-05,
+      "tok_s": 2583
+    },
+    {
+      "step": 30,
+      "loss": 1.7978,
+      "lr": 9.938441702975687e-05,
+      "tok_s": 2579
+    },
+    {
+      "step": 40,
+      "loss": 1.6216,
+      "lr": 9.890738003669028e-05,
+      "tok_s": 2582
+    },
+    {
+      "step": 50,
+      "loss": 1.4679,
+      "lr": 9.829629131445343e-05,
+      "tok_s": 2581
+    },
+    {
+      "step": 60,
+      "loss": 1.6909,
+      "lr": 9.755282581475769e-05,
+      "tok_s": 2578
+    },
+    {
+      "step": 70,
+      "loss": 1.0998,
+      "lr": 9.667902132486009e-05,
+      "tok_s": 2580
+    },
+    {
+      "step": 80,
+      "loss": 1.4509,
+      "lr": 9.567727288213006e-05,
+      "tok_s": 2579
+    },
+    {
+      "step": 90,
+      "loss": 1.4452,
+      "lr": 9.455032620941841e-05,
+      "tok_s": 2577
+    },
+    {
+      "step": 100,
+      "loss": 1.5662,
+      "lr": 9.330127018922195e-05,
+      "tok_s": 2579
+    },
+    {
+      "step": 110,
+      "loss": 1.3704,
+      "lr": 9.193352839727123e-05,
+      "tok_s": 2573
+    },
+    {
+      "step": 120,
+      "loss": 1.0406,
+      "lr": 9.045084971874741e-05,
+      "tok_s": 2573
+    },
+    {
+      "step": 130,
+      "loss": 1.3993,
+      "lr": 8.88572980728486e-05,
+      "tok_s": 2575
+    },
+    {
+      "step": 140,
+      "loss": 1.3269,
+      "lr": 8.715724127386976e-05,
+      "tok_s": 2574
+    },
+    {
+      "step": 150,
+      "loss": 1.462,
+      "lr": 8.535533905932743e-05,
+      "tok_s": 2573
+    },
+    {
+      "step": 160,
+      "loss": 1.2279,
+      "lr": 8.345653031794296e-05,
+      "tok_s": 2575
+    },
+    {
+      "step": 170,
+      "loss": 1.2806,
+      "lr": 8.146601955249193e-05,
+      "tok_s": 2575
+    },
+    {
+      "step": 180,
+      "loss": 1.1756,
+      "lr": 7.938926261462371e-05,
+      "tok_s": 2574
+    },
+    {
+      "step": 190,
+      "loss": 0.918,
+      "lr": 7.723195175075142e-05,
+      "tok_s": 2576
+    },
+    {
+      "step": 200,
+      "loss": 1.3074,
+      "lr": 7.500000000000009e-05,
+      "tok_s": 2575
+    },
+    {
+      "step": 210,
+      "loss": 1.3411,
+      "lr": 7.269952498697741e-05,
+      "tok_s": 2569
+    },
+    {
+      "step": 220,
+      "loss": 1.4813,
+      "lr": 7.033683215379006e-05,
+      "tok_s": 2570
+    },
+    {
+      "step": 230,
+      "loss": 1.3284,
+      "lr": 6.791839747726506e-05,
+      "tok_s": 2570
+    },
+    {
+      "step": 240,
+      "loss": 1.6288,
+      "lr": 6.545084971874744e-05,
+      "tok_s": 2570
+    },
+    {
+      "step": 250,
+      "loss": 1.7878,
+      "lr": 6.294095225512611e-05,
+      "tok_s": 2571
+    },
+    {
+      "step": 260,
+      "loss": 1.6825,
+      "lr": 6.039558454088805e-05,
+      "tok_s": 2571
+    },
+    {
+      "step": 270,
+      "loss": 1.2099,
+      "lr": 5.7821723252011626e-05,
+      "tok_s": 2570
+    },
+    {
+      "step": 280,
+      "loss": 1.3091,
+      "lr": 5.522642316338277e-05,
+      "tok_s": 2571
+    },
+    {
+      "step": 290,
+      "loss": 1.4771,
+      "lr": 5.261679781214727e-05,
+      "tok_s": 2571
+    },
+    {
+      "step": 300,
+      "loss": 1.0913,
+      "lr": 5.000000000000007e-05,
+      "tok_s": 2571
+    },
+    {
+      "step": 310,
+      "loss": 1.5202,
+      "lr": 4.7383202187852876e-05,
+      "tok_s": 2568
+    },
+    {
+      "step": 320,
+      "loss": 1.2693,
+      "lr": 4.477357683661741e-05,
+      "tok_s": 2569
+    },
+    {
+      "step": 330,
+      "loss": 1.1337,
+      "lr": 4.217827674798853e-05,
+      "tok_s": 2569
+    },
+    {
+      "step": 340,
+      "loss": 1.3723,
+      "lr": 3.960441545911209e-05,
+      "tok_s": 2569
+    },
+    {
+      "step": 350,
+      "loss": 1.3359,
+      "lr": 3.7059047744874025e-05,
+      "tok_s": 2569
+    },
+    {
+      "step": 360,
+      "loss": 1.5499,
+      "lr": 3.454915028125268e-05,
+      "tok_s": 2569
+    },
+    {
+      "step": 370,
+      "loss": 1.1808,
+      "lr": 3.208160252273505e-05,
+      "tok_s": 2569
+    },
+    {
+      "step": 380,
+      "loss": 1.389,
+      "lr": 2.9663167846210028e-05,
+      "tok_s": 2570
+    },
+    {
+      "step": 390,
+      "loss": 1.1763,
+      "lr": 2.73004750130227e-05,
+      "tok_s": 2570
+    },
+    {
+      "step": 400,
+      "loss": 1.3025,
+      "lr": 2.500000000000002e-05,
+      "tok_s": 2570
+    },
+    {
+      "step": 410,
+      "loss": 1.1334,
+      "lr": 2.2768048249248692e-05,
+      "tok_s": 2568
+    },
+    {
+      "step": 420,
+      "loss": 1.2506,
+      "lr": 2.0610737385376383e-05,
+      "tok_s": 2568
+    },
+    {
+      "step": 430,
+      "loss": 1.0039,
+      "lr": 1.8533980447508165e-05,
+      "tok_s": 2568
+    },
+    {
+      "step": 440,
+      "loss": 1.6885,
+      "lr": 1.6543469682057116e-05,
+      "tok_s": 2568
+    },
+    {
+      "step": 450,
+      "loss": 1.3592,
+      "lr": 1.4644660940672653e-05,
+      "tok_s": 2568
+    },
+    {
+      "step": 460,
+      "loss": 1.3486,
+      "lr": 1.2842758726130306e-05,
+      "tok_s": 2568
+    },
+    {
+      "step": 470,
+      "loss": 1.279,
+      "lr": 1.1142701927151473e-05,
+      "tok_s": 2569
+    },
+    {
+      "step": 480,
+      "loss": 1.339,
+      "lr": 9.549150281252647e-06,
+      "tok_s": 2569
+    },
+    {
+      "step": 490,
+      "loss": 1.0581,
+      "lr": 8.066471602728818e-06,
+      "tok_s": 2569
+    },
+    {
+      "step": 500,
+      "loss": 1.3058,
+      "lr": 6.69872981077809e-06,
+      "tok_s": 2569
+    },
+    {
+      "step": 510,
+      "loss": 1.5708,
+      "lr": 5.449673790581622e-06,
+      "tok_s": 2567
+    },
+    {
+      "step": 520,
+      "loss": 1.2886,
+      "lr": 4.322727117869972e-06,
+      "tok_s": 2567
+    },
+    {
+      "step": 530,
+      "loss": 1.3237,
+      "lr": 3.3209786751399204e-06,
+      "tok_s": 2568
+    },
+    {
+      "step": 540,
+      "loss": 1.1265,
+      "lr": 2.4471741852423297e-06,
+      "tok_s": 2568
+    },
+    {
+      "step": 550,
+      "loss": 1.3335,
+      "lr": 1.7037086855465882e-06,
+      "tok_s": 2568
+    },
+    {
+      "step": 560,
+      "loss": 1.6331,
+      "lr": 1.0926199633097236e-06,
+      "tok_s": 2568
+    },
+    {
+      "step": 570,
+      "loss": 1.3864,
+      "lr": 6.155829702431183e-07,
+      "tok_s": 2568
+    },
+    {
+      "step": 580,
+      "loss": 1.1831,
+      "lr": 2.7390523158633617e-07,
+      "tok_s": 2568
+    },
+    {
+      "step": 590,
+      "loss": 0.9576,
+      "lr": 6.852326227130851e-08,
+      "tok_s": 2568
+    },
+    {
+      "step": 600,
+      "loss": 0.9617,
+      "lr": 0.0,
+      "tok_s": 2568
+    }
+  ]
+}