mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2025-08-11 02:42:03 +08:00
add latest
This commit is contained in:
parent
b66e483fe8
commit
ee8e5f8ab8
4
api.py
4
api.py
@ -998,9 +998,9 @@ else:
|
|||||||
# 初始化模型
|
# 初始化模型
|
||||||
cnhubert.cnhubert_base_path = cnhubert_base_path
|
cnhubert.cnhubert_base_path = cnhubert_base_path
|
||||||
# tokenizer = AutoTokenizer.from_pretrained(bert_path)
|
# tokenizer = AutoTokenizer.from_pretrained(bert_path)
|
||||||
tokenizer = AutoTokenizer.from_pretrained(os.path.join(os.getcwd(), bert_path), local_files_only=True)
|
tokenizer = AutoTokenizer.from_pretrained(bert_path)
|
||||||
|
|
||||||
bert_model = AutoModelForMaskedLM.from_pretrained(os.path.join(os.getcwd(), bert_path), local_files_only=True)
|
bert_model = AutoModelForMaskedLM.from_pretrained(bert_path)
|
||||||
ssl_model = cnhubert.get_model()
|
ssl_model = cnhubert.get_model()
|
||||||
if is_half:
|
if is_half:
|
||||||
bert_model = bert_model.half().to(device)
|
bert_model = bert_model.half().to(device)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user