mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2025-08-28 12:19:46 +08:00
add all
This commit is contained in:
parent
f623a5bb96
commit
550c37528c
6
api.py
6
api.py
@ -997,8 +997,10 @@ 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)
|
||||||
bert_model = AutoModelForMaskedLM.from_pretrained(bert_path)
|
tokenizer = AutoTokenizer.from_pretrained(os.path.join(os.getcwd(), bert_path))
|
||||||
|
|
||||||
|
bert_model = AutoModelForMaskedLM.from_pretrained(os.path.join(os.getcwd(), 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