Views
No views yet

1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3MODEL_PATH = "double7/Tower-7b-EAX"
4
5tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
6model = AutoModelForCausalLM.from_pretrained(
7 MODEL_PATH, device_map="auto", torch_dtype="auto"
8)
9
10src_lang = "German"
11trg_lang = "Chinese"
12src_text = "Filmkarriere Collinges Filmdebüt in Die kleinen Füchse von 1941 brachte ihr eine Nominierung für den Academy Award als beste Nebendarstellerin ein."
13
14prompt = f"Translate the following text from {src_lang} into {trg_lang}:\n{src_lang}: {src_text}\n{trg_lang}:"
15
16# We use the tokenizer’s chat template to format each message - see https://huggingface.co/docs/transformers/main/en/chat_templating
17messages = [
18 {"role": "user", "content": prompt},
19]
20
21input_text = tokenizer.apply_chat_template(
22 messages, tokenize=False, add_generation_prompt=True
23)
24
25inputs = tokenizer(input_text, return_tensors="pt").to(model.device)
26
27outputs = model.generate(**inputs, do_sample=False, max_new_tokens=256)
28output_text = tokenizer.batch_decode(outputs, skip_special_tokens=False)[0]
29print(output_text)
30# <s><|im_start|> user
31# Translate the following text from German into Chinese:
32# German: Filmkarriere Collinges Filmdebüt in Die kleinen Füchse von 1941 brachte ihr eine Nominierung für den Academy Award als beste Nebendarstellerin ein.
33# Chinese:<|im_end|>
34# <|im_start|> assistant
35# 电影生涯 科林格的电影处女作《小狐狸》于 1941 年上映,她因此获得了奥斯卡最佳女配角提名。<|im_end|>
361prompt1 = f"Translate the following text from {src_lang} into {trg_lang}:\n{src_lang}: {src_text}\n{trg_lang}:"
2
3prompt1 = f"Please provide a translation from {src_lang} to {trg_lang} for the following text:\n{src_text}\nTarget:",
4
5prompt2 = f"Translate this {src_lang} text into {trg_lang}:\nSource: {src_text}\nTranslation:",prompt1 for the evaluation.ChatML prompt templates without any system prompts. An example follows below:<|im_start|>user
{USER PROMPT}<|im_end|>
<|im_start|>assistant
{MODEL RESPONSE}<|im_end|>
<|im_start|>user
[...]1@misc{yang2025enanchoredx2xenglishanchoredoptimizationmanytomany,
2 title={EnAnchored-X2X: English-Anchored Optimization for Many-to-Many Translation},
3 author={Sen Yang and Yu Bao and Yu Lu and Jiajun Chen and Shujian Huang and Shanbo Cheng},
4 year={2025},
5 eprint={2509.19770},
6 archivePrefix={arXiv},
7 primaryClass={cs.CL},
8 url={https://arxiv.org/abs/2509.19770},
9}