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

Refactor: run "hatch run style:fmt"

Browse Source
This commit is contained in:
tsukumi
2024-10-25 22:18:57 +09:00
parent d762bf4401
commit 375eb7f3a9
2 changed files with 17 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
style_bert_vits2/tts_model.py
View File

@@ -150,7 +150,9 @@ class TTSModel:
hps=self.hyper_parameters, hps=self.hyper_parameters,
) )
# 愚直。もっと上手い方法ありそう # 愚直。もっと上手い方法ありそう
params = zip(self.net_g.dec.parameters(), null_model_add.dec.parameters()) params = zip(
self.net_g.dec.parameters(), null_model_add.dec.parameters()
)
for v in params: for v in params:
v[0].data.add_(v[1].data, alpha=float(null_model_info["weight"])) v[0].data.add_(v[1].data, alpha=float(null_model_info["weight"]))
params = zip( params = zip(
@@ -165,14 +167,18 @@ class TTSModel:
for v in params: for v in params:
v[0].data.add_(v[1].data, alpha=float(null_model_info["style"])) v[0].data.add_(v[1].data, alpha=float(null_model_info["style"]))
# テンポは sdp と dp 二つあるからとりあえずどっちも足す # テンポは sdp と dp 二つあるからとりあえずどっちも足す
params = zip(self.net_g.sdp.parameters(), null_model_add.sdp.parameters()) params = zip(
self.net_g.sdp.parameters(), null_model_add.sdp.parameters()
)
for v in params: for v in params:
v[0].data.add_(v[1].data, alpha=float(null_model_info["tempo"])) v[0].data.add_(v[1].data, alpha=float(null_model_info["tempo"]))
params = zip(self.net_g.dp.parameters(), null_model_add.dp.parameters()) params = zip(self.net_g.dp.parameters(), null_model_add.dp.parameters())
for v in params: for v in params:
v[0].data.add_(v[1].data, alpha=float(null_model_info["tempo"])) v[0].data.add_(v[1].data, alpha=float(null_model_info["tempo"]))
logger.info(f"Null models merged successfully ({time.time() - start_time:.2f}s)") logger.info(
f"Null models merged successfully ({time.time() - start_time:.2f}s)"
)
# ONNX 推論時 # ONNX 推論時
else: else: