from transformers import AutoTokenizer, AutoModelForCausalLMimport torch model_id = "gateremark/kikuyu_translategemma_12b_merged_V2" tokenizer = AutoTokenizer.from_pretrained(model_id)model = AutoModelForCausalLM.from_pretrained( model_id, torch_dtype=torch.bfloat16, device_map="auto",) def translate_to_kikuyu(text: str) -> str: messages = [ { "role": "user", "content": [{ "type": "text", "source_lang_code": "en", "target_lang_code": "ki", "text": text }] } ] input_ids = tokenizer.apply_chat_template( messages, tokenize=True, add_generation_prompt=True, return_tensors="pt" ).to(model.device) terminators = [ tokenizer.eos_token_id, tokenizer.convert_tokens_to_ids("<end_of_turn>"), ] with torch.no_grad(): outputs = model.generate( input_ids=input_ids, max_new_tokens=256, temperature=0.3, do_sample=True, eos_token_id=terminators, ) response = tokenizer.decode( outputs[0][input_ids.shape[1]:], skip_special_tokens=True ) return response.strip() # Exampleprint(translate_to_kikuyu("Hello, how are you?"))# Output: Hihi, ũrĩ atĩa?