OSS AI event extractor (7B)

Fine-tune of Qwen/Qwen2.5-7B-Instruct, quantised to Q4_K_M, that pulls structured event data out of free-form Discord messages for the Vaktovian Systems Department.

What it does

Given a message like:

SURGE @ VAK Eular Station LOSS

it returns:

{"CLAN": "The Imperial Insurgence", "BASE": "Eular Station", "TYPE": "Defense", "RESULT": "Loss"}

Only those four fields. The date, the author and the screenshot come off the Discord message object instead, because those were never a language problem and asking a model for them just adds somewhere for it to be wrong.

Results

On the project's earlier validation set this model scored 92.7% exact match where the 1.5B fine-tune scored 93.6%. Same training data, same recipe, 4.7x the size, no gain, so the 1.5B is the production model and this one is kept for the record. It has not been re-scored on the newer 892-row benchmark.

Training

  • 8,020 examples, 2 epochs, LoRA r=16 alpha=32 on all attention and MLP projections
  • Loss masked to the answer only, so it is not trained on the prompt or on padding
  • Dynamic padding rather than a fixed length, the median example is 209 tokens

Usage

from llama_cpp import Llama

llm = Llama(model_path="oss_ai_7b_q4.gguf", n_ctx=512)
out = llm.create_chat_completion(
    messages=[
        {"role": "system", "content": SYSTEM_PROMPT},
        {"role": "user", "content": "SURGE @ VAK Eular Station LOSS"},
    ],
    temperature=0,
)

The system prompt has to match the one it was trained on, which is in utils/constants.py in the bot repo. A different prompt will make it noticeably worse.

Downloads last month
29
GGUF
Model size
8B params
Architecture
qwen2
Hardware compatibility
Log In to add your hardware

We're not able to determine the quantization variants.

Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Model tree for constmix/oss-ai-7b-gguf

Base model

Qwen/Qwen2.5-7B
Quantized
(418)
this model