Refactor: minor adjustments

2024-03-07 03:54:02 +00:00
parent 62919e904e
commit 4f11b011fd
8 changed files with 18 additions and 33 deletions
--- a/style_bert_vits2/text_processing/japanese/bert_feature.py
+++ b/style_bert_vits2/text_processing/japanese/bert_feature.py
@@ -8,7 +8,7 @@ from style_bert_vits2.text_processing import bert_models
 from style_bert_vits2.text_processing.japanese.g2p import text_to_sep_kata


-models: dict[str, PreTrainedModel] = {}
+models: dict[torch.device | str, PreTrainedModel] = {}


 def extract_bert_feature(