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

Fix: Specifying the device_map option in extract_bert_feature()

Browse Source
This commit is contained in:
tsukumi
2024-08-11 08:00:56 +09:00
parent 80dd6dbc22
commit 0e71d7ef7c
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -29,7 +29,7 @@ def extract_bert_feature(
if device == "cuda" and not torch.cuda.is_available():
device = "cpu"
model = bert_models.load_model(Languages.ZH)
model = bert_models.load_model(Languages.ZH, device_map=device)
bert_models.transfer_model(Languages.ZH, device)
style_res_mean = None

2
style_bert_vits2/nlp/english/bert_feature.py
View File

@@ -29,7 +29,7 @@ def extract_bert_feature(
if device == "cuda" and not torch.cuda.is_available():
device = "cpu"
model = bert_models.load_model(Languages.EN)
model = bert_models.load_model(Languages.EN, device_map=device)
bert_models.transfer_model(Languages.EN, device)
style_res_mean = None

2
style_bert_vits2/nlp/japanese/bert_feature.py
View File

@@ -36,7 +36,7 @@ def extract_bert_feature(
if device == "cuda" and not torch.cuda.is_available():
device = "cpu"
model = bert_models.load_model(Languages.JP)
model = bert_models.load_model(Languages.JP, device_map=device)
bert_models.transfer_model(Languages.JP, device)
style_res_mean = None