Views
No views yet
| Property | Value |
|---|---|
| Base Model | savinugunarathna/Gemma3-Singlish-Sinhala-Merged |
| Fine-tuning Method | QLoRA (4-bit, r=16) |
| Upload Type | merged |
| Task | Code-mixed → Sinhala transliteration |
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3import torch
4
5MODEL_ID = "Pudamya/Gemma3-Singlish-Codemix"
6
7tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
8
9model = AutoModelForCausalLM.from_pretrained(MODEL_ID, torch_dtype=torch.float16, device_map='auto')
10
11def translate(text):
12 prompt = (
13 "### Instruction:\n"
14 "Convert the following code-mixed Singlish-English sentence into proper Sinhala script.\n\n"
15 f"### Input:\n{text}\n\n"
16 "### Response:\n"
17 )
18 inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
19 with torch.no_grad():
20 out = model.generate(
21 **inputs,
22 max_new_tokens=150,
23 do_sample=False,
24 repetition_penalty=1.1,
25 pad_token_id=tokenizer.eos_token_id,
26 )
27 decoded = tokenizer.decode(out[0], skip_special_tokens=True)
28 return decoded.split("### Response:")[-1].strip()
29
30print(translate("mama api ekka movie eke gihin fun hari thibba"))