Views
No views yet
1from transformers import T5ForConditionalGeneration, T5Tokenizer
2
3device = 'cuda' #or 'cpu' for translate on cpu
4
5model_name = 'utrobinmv/t5_translate_en_ru_zh_large_1024'
6model = T5ForConditionalGeneration.from_pretrained(model_name)
7model.to(device)
8tokenizer = T5Tokenizer.from_pretrained(model_name)
9
10prefix = 'translate to zh: '
11src_text = prefix + "Съешь ещё этих мягких французских булок."
12
13# translate Russian to Chinese
14input_ids = tokenizer(src_text, return_tensors="pt")
15
16generated_tokens = model.generate(**input_ids.to(device))
17
18result = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
19print(result)
20# 再吃这些法国的甜蜜的面包。1from transformers import T5ForConditionalGeneration, T5Tokenizer
2
3device = 'cuda' #or 'cpu' for translate on cpu
4
5model_name = 'utrobinmv/t5_translate_en_ru_zh_large_1024'
6model = T5ForConditionalGeneration.from_pretrained(model_name)
7model.to(device)
8tokenizer = T5Tokenizer.from_pretrained(model_name)
9
10prefix = 'translate to ru: '
11src_text = prefix + "再吃这些法国的甜蜜的面包。"
12
13# translate Russian to Chinese
14input_ids = tokenizer(src_text, return_tensors="pt")
15
16generated_tokens = model.generate(**input_ids.to(device))
17
18result = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
19print(result)
20# Съешьте этот сладкий хлеб из Франции.