Fix: spm.model is missing

style_bert_vits2/nlp/onnx_bert_models.py

@@ -87,6 +87,15 @@ def load_model(
                 revision=revision,
             )
         )
+        # 英語用 BERT のみ、spm.model もダウンロードする
+        # Fast 版の BERT トークナイザーでは不要なはずだが、念のため
+        if language == Languages.EN:
+            hf_hub_download(
+                repo_id=pretrained_model_name_or_path,
+                filename="spm.model",
+                cache_dir=cache_dir,
+                revision=revision,
+            )
     # pretrained_model_name_or_path にファイルパスが指定された場合:
     # 既にダウンロード済みという前提のもと、モデルへのローカルパスを model_path に格納する
     else:
@@ -138,7 +147,7 @@ def load_tokenizer(
         revision (str): モデルの Hugging Face 上の Git リビジョン。指定しない場合は最新の main ブランチの内容が利用される (デフォルト: None)
 
     Returns:
-        Union[PreTrainedTokenizer, PreTrainedTokenizerFast, DebertaV2Tokenizer]: ロード済みの BERT トークナイザー
+        Union[PreTrainedTokenizer, PreTrainedTokenizerFast, DebertaV2TokenizerFast]: ロード済みの BERT トークナイザー
     """
 
     # すでにロード済みの場合はそのまま返す