lbourdois's picture
Update model card for Malay
0ebfbaf verified
|
Raw
History Blame Contribute Delete
3.09 kB
metadata
pipeline_tag: text-generation
language: msa
license: gemma
tags:
  - trimmed
library_name: transformers
base_model: google/gemma-3-1b-it
base_model_relation: quantized
datasets:
  - lbourdois/fineweb-2-trimming

gemma-3-1b-it-msa-16384

This model is a 28.31% smaller version of google/gemma-3-1b-it optimized for Malay language via vocabulary size reduction using the trimming method.
This trimmed model should perform similarly to the original model with only 16,384 tokens and a much smaller memory footprint. However, it may not perform well for other languages as tokens not commonly used in the selected languages were removed from the vocabulary.

Model Statistics

Metric Original Trimmed Reduction
Vocabulary size 262,144 tokens 16,384 tokens 93.75%
Model size 999,885,952 params 716,770,432 params 28.31%

image

Mining Dataset Statistics

Usage

from transformers import AutoModelForCausalLM, AutoTokenizer

model_name = "alphaedge-ai/gemma-3-1b-it-msa-16384"

tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(
    model_name,
    torch_dtype="auto",
    device_map="auto"
)

prompt = "Your prompt in Malay."
messages = [{"role": "user", "content": prompt}]
text = tokenizer.apply_chat_template(
    messages,
    tokenize=False,
    add_generation_prompt=True
)
model_inputs = tokenizer([text], return_tensors="pt").to(model.device)

generated_ids = model.generate(**model_inputs, max_new_tokens=256)
output_ids = generated_ids[0][len(model_inputs.input_ids[0]):]
response = tokenizer.decode(output_ids, skip_special_tokens=True)
print(response)

Citations

Gemma 3

@misc{gemmateam2025gemma3technicalreport,
      title={Gemma 3 Technical Report},
      author={Gemma Team},
      year={2025},
      eprint={2503.19786},
      archivePrefix={arXiv},
      primaryClass={cs.CL},
      url={https://arxiv.org/abs/2503.19786},
}

Trimming blog post

@misc{hf_blogpost_trimming,
      title={Introduction to Trimming}, 
      author={Loïck BOURDOIS and Tom AARSEN and Bram VANROY and Christopher AKIKI and Woojun JUNG and Manuel ROMERO and Prithiv SAKTHI},
      year={2026},
      url={https://huggingface.co/blog/lbourdois/introduction-to-trimming}, 
}

License

This model is derived from google/gemma-3-1b-it. Use of this model is governed by the Gemma Terms of Use. By using this model, you agree to the Gemma Terms of Use. This model is not affiliated with or endorsed by Google.