Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "dadu/qwen3-0.6b-translation-synthetic-reasoning-1"
4model = AutoModelForCausalLM.from_pretrained(model_name)
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6
7# Few-shot examples provide context for the translation style
8few_shot_prompt = """
9Examples:
10
11source: Aŋkɛ bímbɔ áwúlégé, ɛkiɛ́nné Ɛsɔwɔ ɛ́kwɔ́ Josɛf ushu né gejya, ɛ́jɔɔ́ ne ji ɛké...
12target: Jalla nekztanaqui nii magonacaz̈ ojktan tsjii Yooz Jilirz̈ anjilaqui wiiquin Josez̈quiz parisisquichic̈ha...
13
14source: Josɛf ápégé, asɛ maá yimbɔ ne mmá wuú áfɛ́ né mme Isrɛli.
15target: Jalla nuz̈ cjen Josequi z̈aaz̈cu Israel yokquin nii uztan maatan chjitchic̈ha.
16
17Query: ɛké “Josɛf, kwilé ka ɔ́kpá maá yina ne mma wuú, ɛnyú dékéré meso né mme Isrɛli. Bɔɔ́ abi ákɛlege manwá ji ágboó.”
18"""
19
20messages = [
21 {"role": "system", "content": "You are a helpful Bible translation assistant. Given examples of language pairs and a query, you will write a high quality translation with reasoning."},
22 {"role": "user", "content": few_shot_prompt}
23]
24
25text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
26inputs = tokenizer(text, return_tensors="pt")
27
28outputs = model.generate(**inputs, max_new_tokens=1024, temperature=0.7)
29response = tokenizer.decode(outputs[0], skip_special_tokens=True)
30print(response)dadu/translation-synthetic-reasoning-1