diff --git a/README.md b/README.md index 0ae3ca8..42bb339 100644 --- a/README.md +++ b/README.md @@ -1,2 +1,14 @@ # sbv2-api -Only infer tts +このプロジェクトはStyle-Bert-ViTS2をONNX化したものをRustで実行するのを目的としています。つまり推論しか行いません。 + +学習したいのであれば、Style-Bert-ViT2で調べてやってください。 + +## ONNX化する方法 +dabertaとstbv2本体をonnx化する必要があります。 + +あくまで推奨ですが、onnxsimを使うことをお勧めします。 +onnxsim使うことでモデルのサイズを軽くすることができます。 + +## onnxモデルの配置方法 +- `models/daberta.onnx` - DaBertaのonnxモデル +- `models/sbv2.onnx` - `Style-Bert-ViT2`の本体