0

機械翻訳モデルを微調整し、トレーニング中に保存された pytorch_model.bin モデル チェックポイントを読み込んで単語の翻訳を予測しようとしています。Transformers.modeling_outputs.Seq2SeqModelOutput から通常のテキストに変換するにはどうすればよいですか?

model = AutoModel.from_pretrained('/content/drive/MyDrive/model', cache_dir=None)

 tokenizer = AutoTokenizer.from_pretrained('/content/drive/MyDrive/model', cache_dir=None)

model.eval()

inputs2 = tokenizer('word', return_tensors="pt")["input_ids"]

inputs2.data #results in =>
tensor([[ 1415,   259, 54622,     1]])

outputs = model(input_ids=inputs2, decoder_input_ids=inputs2)
type(outputs) #results in =>transformers.modeling_outputs.Seq2SeqModelOutput

**output_str = #translation of the input word**
4

0 に答える 0