--- library_name: transformers tags: - trl - sft license: apache-2.0 datasets: - interstellarninja/hermes_reasoning_tool_use language: - en metrics: - accuracy base_model: - google/functiongemma-270m-it pipeline_tag: text-generation --- # FunctionGemma Hermes Tool-Use (3K Fine-tuned) This model is a **fine-tuned version of Google’s FunctionGemma (270M)**, trained on a curated subset of the **Hermes Tool-Use** dataset to improve **structured function calling**. The goal of this fine-tuning is **higher accuracy and reliability** when selecting the correct tool and emitting a valid function call in the expected format. > [!Note] > **Check out Fine-tuning script:** [https://www.kaggle.com/code/kingabzpro/finetuning-functiongemma](https://www.kaggle.com/code/kingabzpro/finetuning-functiongemma) ## 🚀 What’s Improved Evaluation was run on a held-out validation set (50 examples): | Metric | Before FT | After FT | |------|-----------|----------| | Tool Selection Accuracy | **88.0%** | **98.0%** | | Absolute Gain | – | **+10.0%** | This shows the model learns **better tool selection and call consistency**, even though the base model already performs strongly. ## 🧠 Supported Output Format The model emits function calls in **FunctionGemma-style** tags: ```text call:tool_name{args:{...}} ```` This is compatible with downstream tool execution pipelines. ## 📦 Installation ```bash pip install transformers accelerate sentencepiece ``` ## 🔧 Usage Example (Function Calling) ```python from transformers import AutoProcessor, AutoModelForCausalLM repo_id = "kingabzpro/functiongemma-hermes-3k-ft" processor = AutoProcessor.from_pretrained(repo_id) model = AutoModelForCausalLM.from_pretrained(repo_id) # Tool definition (HF function schema) tools = [ { "type": "function", "function": { "name": "billboard_global_200", "description": "Fetch Billboard Global 200 chart information for a specific date.", "parameters": { "type": "object", "properties": { "date": { "type": "string", "description": "Date in YYYY-MM-DD format", "default": "2020-09-19", } }, "required": ["date"], }, }, } ] messages = [ { "role": "developer", "content": ( "You are a function calling AI model. " "Each function call must be enclosed in XML tags." ), }, { "role": "user", "content": ( "Which songs were at positions 1, 11, 21, 31, and 41 " "on the Billboard Global 200 chart, and who sang them?" ), }, ] inputs = processor.apply_chat_template( messages, tools=tools, add_generation_prompt=True, return_dict=True, return_tensors="pt", ) inputs = {k: v.to(model.device) for k, v in inputs.items()} outputs = model.generate( **inputs, max_new_tokens=256, pad_token_id=processor.eos_token_id, ) gen = processor.decode( outputs[0][inputs["input_ids"].shape[-1]:], skip_special_tokens=True, ) print(gen) ``` ### ✅ Example Output ```text call:billboard_global_200{args:{"date": "2006-03-20"}} ``` ## 🎯 Intended Use * Tool / function calling research * Agent systems and planners * Structured API invocation * Evaluation of tool-selection accuracy * Lightweight function-calling demos (CPU / small GPU friendly) ## ⚠️ Limitations * Trained on a **subset (3K)** of Hermes Tool-Use data * Focused on **tool selection**, not long-form reasoning * Not instruction-tuned for general chat beyond tool use ## 📜 Attribution * Base model: **Google FunctionGemma** * Dataset: **Hermes Tool-Use** * Fine-tuning & evaluation: **kingabzpro**