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_small_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_small_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#Цель разработки - предоставить персонализированный синхронный перевод для пользователей.