| --- |
| license: mpl-2.0 |
| datasets: |
| - acon96/Home-Assistant-Requests-V2 |
| - TitleOS/HomeAssistant-Loki-Personality |
| language: |
| - en |
| base_model: |
| - Qwen/Qwen3.5-2B |
| tags: |
| - HomeAssistant |
| - HomeAutomation |
| --- |
| # LokiHA-2B (FP32 Merged) |
|
|
| LokiHA-2B is a 2 billion parameter causal language model designed to bridge complex Home Assistant tool-calling with a distinct, consistent character personality. This repository contains the raw, unquantized FP32 merged model weights. |
|
|
| This model is built on the `Qwen/Qwen3.5-2B` base and trained to strictly adhere to the ChatML prompt format for both conversational turns and JSON tool emissions. |
|
|
| ## Training Data & Methodology |
|
|
| We faced a classic mixing problem during fine-tuning: balancing a massive functional dataset against a highly specific, smaller personality dataset. To prevent the tool-calling logic from drowning out the character, this model was trained using a calculated upsampling strategy. |
|
|
| The final training mixture consists of: |
| * **85% Functionality:** `acon96/Home-Assistant-Requests-V2` (Truncated to 50,000 rows). This teaches the model the rigorous JSON schema required to trigger Home Assistant services, devices, and scripts. |
| * **15% Personality:** `TitleOS/HomeAssistant-Loki-Personality`. This dataset was dynamically upsampled to represent roughly 15% of the total training steps, ensuring the "Loki" persona remains present and stable without breaking the underlying syntax required for home automation. |
|
|
| ## Prompt Format |
|
|
| This model requires the standard **ChatML** format to function correctly. Tool definitions must be injected into the system prompt. |
|
|
| ```text |
| <|im_start|>system |
| You are Loki, a helpful AI Assistant that controls the devices in a house using the Home Assistant platform. |
| [Insert Tools/State JSON here]<|im_end|> |
| <|im_start|>user |
| Turn off the living room lights.<|im_end|> |
| <|im_start|>assistant |
| ``` |
|
|
| ## License |
|
|
| This model is distributed under a **modified Mozilla Public License 2.0 (MPL 2.0) with a Common Clause**. |
|
|
| Please see the `license.md` file included in this repository for the exact legal text and restrictions regarding commercial use and distribution. |
