from transformers import AutoModelForCausalLM, AutoTokenizer
model_name = "Respair/Hayate_Translate_base_v1.0_JPtoEN"
tokenizer = "Qwen/Qwen3-4B-Instruct-2507"
tokenizer = AutoTokenizer.from_pretrained(tokenizer)
model = AutoModelForCausalLM.from_pretrained(
model_name,
torch_dtype="auto",
device_map="auto"
)
text = (
# text
).replace('\n', ' ').strip()
prompt = "JP->EN: "
messages = [
{"role": "user", "content": prompt + text}
]
text = tokenizer.apply_chat_template(
messages,
tokenize=False,
add_generation_prompt=True,
)
model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
generated_ids = model.generate(
**model_inputs,
max_new_tokens=8192,
do_sample=True,
temperature=0.1,
top_p=0.95,
)
output_ids = generated_ids[0][len(model_inputs.input_ids[0]):].tolist()
content = tokenizer.decode(output_ids, skip_special_tokens=True)
print(content)
Details
For more details, please visit Respair/Hayate_Translate_FT_EN2JP
- Downloads last month
- 41