metadata
license: gemma
language:
- te
base_model: google/gemma-2-9b
library_name: transformers
Gemma2 9B for Telugu: 100 target vocabulary size + Mean target vocabulary initialization + 2x2LS/MTP/512 training
This model is built on top of Gemma2 9B adapted for Telugu using 30K target language sentences sampled from CC-100.
Model Details
- Vocabulary: This model has an additional 100 target vocabulary.
- Target vocabulary initialization: The target weights of the embedding were initialized using Mean initialization.
- Training: This model was additionally pre-trained on 30K target language sentences sampled from CC-100. The training was conducted with the 2x2LS/MTP/512 strategies introduced in the paper.
Model Description
- Language: Telugu
- License: Gemma Terms of Use
- Fine-tuned from model: google/gemma-2-9b
Model Sources
- Repository: https://github.com/gucci-j/lowres-cve
- Paper: https://arxiv.org/abs/2406.11477
How to Get Started with the Model
Use the code below to get started with the model.
from transformers import AutoTokenizer, AutoModelForCausalLM
model = AutoModelForCausalLM.from_pretrained(
"atsuki-yamaguchi/gemma-2-9b-te-30K-mean"
)
tokenizer = AutoTokenizer.from_pretrained(
"atsuki-yamaguchi/gemma-2-9b-te-30K-mean"
)