Mojicast punct-onnx — 日本語句読点復元モデル(ONNX変換版)

配信用リアルタイム字幕アプリ Mojicast が使用する、 日本語テキストへの句読点(、。)復元モデルの ONNX 変換版です。

元モデル(クレジット)

各制作者に感謝します。

変更内容

  • 上記2つを結合した PyTorch モデル(BERT + 線形層のトークン分類)を ONNX(opset 17 / fp32 / 単一ファイル)へエクスポート
  • 変換スクリプト: Mojicast リポジトリの tools/convert_models.py

ファイル

ファイル 内容
punct_bert.onnx モデル本体(fp32・約364MB)
vocab.txt 文字語彙(tohoku-nlp BERT char v3 と同一)

入出力

  • 入力: input_ids / attention_mask(int64、文字単位トークン)
  • 出力: logits(各トークン直後の [読点, 句点] のロジット。sigmoid して閾値判定)

ライセンス

Apache-2.0(元モデルに従う)

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for ishiki-emo/mojicast-punct-onnx

Quantized
(1)
this model