Views
No views yet
1from transformers import MarianMTModel, MarianTokenizer
2
3device = torch.device('cuda')
4model_name = 'cbdb/ClassicalChineseOfficeTitleTranslation'
5tokenizer = MarianTokenizer.from_pretrained(model_name)
6model = MarianMTModel.from_pretrained(model_name).to(device)1# Load your data here
2tobe_translated = ['講筵官','判司簿尉','散騎常侍','殿中省尚輦奉御']1inputs = tokenizer(tobe_translated, return_tensors="pt", padding=True).to(device)
2translated = model.generate(**inputs, max_length=128)
3tran = [tokenizer.decode(t, skip_special_tokens=True) for t in translated]
4for c, t in zip(tobe_translated, tran):
5 print(f'{c}: {t}')