Downloads · 30 days
24
7% of all-time downloads
umisato/mt5-base-kirakira-names
mt5-base-kirakira-names is a text generation model from umisato. Use it when you need the model to write or continue text. It is set up for transformers.
このモデルはキラキラネームの読み方を予測するために、mT5モデルをファインチューニングをしたものです。 読み方を予測するため、入力は漢字である必要があります。
Downloads · 30 days
24
7% of all-time downloads
All-time downloads
354
Public
Parameters
582M
7 GB on disk
Likes
1
Public
Click a slice to open those files.
.pt4.7 GB · 66%
From the Hugging Face model README
このモデルはキラキラネームの読み方を予測するために、mT5モデルをファインチューニングをしたものです。 読み方を予測するため、入力は漢字である必要があります。
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("umisato/mt5-base-kirakira-names")
model = AutoModelForSeq2SeqLM.from_pretrained("umisato/mt5-base-kirakira-names")
inputs = tokenizer("your-input(kanji)", return_tensors="pt")
outputs = model.generate(**inputs)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))