Attribute name incorrect
#16
by LiPengtao12138 - opened
Only src_lang can be used to refer to the target language, using tgt_lang is invalid.
def t2t_small100(text, source_language = None, target_language = None):
model_small100_tokenizer.src_lang = target_language
# model_small100_tokenizer.tgt_lang = target_language # invalid
encoded_text = model_small100_tokenizer(text, return_tensors="pt")
generated_tokens = model_small100.generate(**encoded_text)
predict_text = model_small100_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]