Beam IME 0.6B (bf16)

The full-precision weights of the model behind Beam IME, a local Chinese input method for Fcitx5-Rime and Weasel. The model reads the keys you typed (full pinyin, initials, mixed syllable prefixes and complete English words) and writes the text.

  • Base: Qwen/Qwen3-0.6B-Base, full-parameter SFT
  • Internal name: r5-v2ex (one epoch, 3.84M examples, max length 128 tokens, lr 2e-5)
  • Weights: bf16 safetensors; input and output embeddings are tied
  • Prompt: keys_llm_v1
  • License: Apache-2.0

The Q8_0 GGUF shipped with Beam (beam-0.6b-q8_0.gguf, SHA-256 35eccd83…7203309c, in the v0.1.0-beta.1 release) is quantized from exactly these weights: re-quantizing every tensor of this checkpoint to Q8_0 reproduces that file's tensors byte for byte. Use the GGUF for the input method itself; this checkpoint is for further training, evaluation, or applying the Beam LoRA adapters with PEFT.

Prompt

上文:<up to 64 recently committed characters, line omitted when empty>
按键:<keys, one letter per token separated by spaces>
结果:<text><|endoftext|>
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer

tok = AutoTokenizer.from_pretrained("7eu7d7/beam-ime-0.6b")
model = AutoModelForCausalLM.from_pretrained("7eu7d7/beam-ime-0.6b", dtype=torch.bfloat16)
prompt = "上文:今晚\n按键:" + " ".join("wmqcfb") + "\n结果:"
ids = tok(prompt, return_tensors="pt").input_ids
out = model.generate(ids, max_new_tokens=16, num_beams=4, num_return_sequences=3, do_sample=False,
                     pad_token_id=tok.eos_token_id)
print([tok.decode(o[ids.shape[1]:], skip_special_tokens=True) for o in out])
# ['我们去吃饭吧', '我们去吃饭不', '我们去吃饭呗']

Other outputs from this checkpoint: nihao → 你好, mingtianjian → 明天见, yonglinux → 用Linux.

Adapters

  • English translation: translate-en-v1, a LoRA (r128) trained on this checkpoint with prompt keys_llm_v4. The release has both a GGUF for Beam and the PEFT checkpoint; its adapter_config.json names the base beam-0.6b-r5-v2ex, which is this repository.

Training data

LCCC, GeneInput, V2EX, Rime vocabulary, and chat and vocabulary data provided by the maintainer, with synthetic typing variants (full spelling, initials, mixed prefixes, English words). GeneInput is published without a license; LCCC is MIT. The training data is not released.

Limitations

Initials are ambiguous and the model picks one reading; it can produce unexpected colloquial choices. Sequence scores are not confidence estimates; an input method shows several candidates and the user chooses. Inputs are limited to 40 key letters in Beam.

Downloads last month
20
Safetensors
Model size
0.8B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for 7eu7d7/beam-ime-0.6b

Finetuned
(738)
this model