Views
No views yet
asl_GL.asl_GL (warm-started from en_XX)| Metric | English → ASL Gloss | ASL Gloss → English |
|---|---|---|
| chrF | 28.02 | 39.26 |
| BLEU | 3.03 | 6.91 |
| ROUGE-L | 25.72 | 30.94 |
| Token-F1 | 31.88 | 36.88 |
1from transformers import MBart50TokenizerFast, MBartForConditionalGeneration
2
3REPO = "manohonsy/asl-mbart-50-lora"
4tokenizer = MBart50TokenizerFast.from_pretrained(REPO)
5model = MBartForConditionalGeneration.from_pretrained(REPO)
6
7# English → ASL gloss
8tokenizer.src_lang = "en_XX"
9inputs = tokenizer("I want to bake a chocolate cake for my sister's birthday.",
10 return_tensors="pt", max_length=128, truncation=True)
11out = model.generate(
12 **inputs,
13 forced_bos_token_id=tokenizer.convert_tokens_to_ids("asl_GL"),
14 max_length=128,
15 num_beams=4,
16)
17print(tokenizer.decode(out[0], skip_special_tokens=True))
18# Expected: ASL gloss like "IX WANT BAKE CHOCOLATE CAKE FOR SISTER BIRTHDAY"
19
20# ASL gloss → English
21tokenizer.src_lang = "asl_GL"
22inputs = tokenizer("IX WANT BAKE CHOCOLATE CAKE FOR SISTER BIRTHDAY",
23 return_tensors="pt", max_length=128, truncation=True)
24out = model.generate(
25 **inputs,
26 forced_bos_token_id=tokenizer.convert_tokens_to_ids("en_XX"),
27 max_length=128,
28 num_beams=4,
29)
30print(tokenizer.decode(out[0], skip_special_tokens=True))asl_GL in lang_code_to_id:1asl_id = tokenizer.convert_tokens_to_ids("asl_GL")
2tokenizer.lang_code_to_id["asl_GL"] = asl_idasl_GL embedding row updates during training# for fingerspelled/acronyms, cl: for classifier predicates, and IX for pointing.