indiejoseph/yue-zh-translation
Viewer • Updated • 212k • 65 • 11
How to use hou000123/zh2yue-translation with Transformers:
# Use a pipeline as a high-level helper
# Warning: Pipeline type "translation" is no longer supported in transformers v5.
# You must load the model directly (see below) or downgrade to v4.x with:
# 'pip install "transformers<5.0.0'
from transformers import pipeline
pipe = pipeline("translation", model="hou000123/zh2yue-translation") # Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("hou000123/zh2yue-translation")
model = AutoModelForSeq2SeqLM.from_pretrained("hou000123/zh2yue-translation", device_map="auto")Translation Chinese To Yue (Cantonese)
中文翻譯粵語
pip install datasets transformers torch sentencepiece
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
model_name = "hou000123/zh2yue-translation"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
def translate_to_cantonese(text):
inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
outputs = model.generate(**inputs)
return tokenizer.decode(outputs[0], skip_special_tokens=True)
print(translate_to_cantonese("今天天气很好")) # 输出: "今日天氣好好"```
Base model
Helsinki-NLP/opus-mt-zh-en