Views
No views yet
v6-ksl-prompt) into Qwen/Qwen2.5-0.5B-Instruct (0.5B parameters, 25% of the 2B teacher).| Model | Token F1 | Exact Match |
|---|---|---|
| Teacher (Gemma2-2B v6) | 0.616 | 14.0% |
| This model | 0.604 | 15.5% |
Translate the following sentence into Kenyan Sign Language (KSL) glosses.
<English sentence>1import torch
2from peft import PeftModel
3from transformers import AutoModelForCausalLM, AutoTokenizer
4
5BASE = "Qwen/Qwen2.5-0.5B-Instruct"
6ADAPTER = "SignvrseOfficial/Glosser_Qwen25_0.5B_it_v1"
7
8tokenizer = AutoTokenizer.from_pretrained(ADAPTER)
9base = AutoModelForCausalLM.from_pretrained(
10 BASE, torch_dtype=torch.float16, device_map="auto"
11)
12model = PeftModel.from_pretrained(base, ADAPTER)
13model.eval()
14
15sentence = "That house is ours."
16messages = [{
17 "role": "user",
18 "content": (
19 "Translate the following sentence into Kenyan Sign Language "
20 f"(KSL) glosses.\n\n{sentence}"
21 ),
22}]
23prompt = tokenizer.apply_chat_template(
24 messages, tokenize=False, add_generation_prompt=True
25)
26inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
27out = model.generate(**inputs, max_new_tokens=128, do_sample=False)
28gloss = tokenizer.decode(out[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True)
29print(gloss)