tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Refactor: minor adjustments

Browse Source
This commit is contained in:
tsukumi
2024-03-07 03:54:02 +00:00
parent 62919e904e
commit 4f11b011fd
8 changed files with 18 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
style_bert_vits2/text_processing/chinese/bert_feature.py
View File

@@ -7,7 +7,7 @@ from style_bert_vits2.constants import Languages
from style_bert_vits2.text_processing import bert_models
models: dict[str, PreTrainedModel] = {}
models: dict[torch.device | str, PreTrainedModel] = {}
def extract_bert_feature(