Update HubertSoft.py
This commit is contained in:
parent
eae42c0dcd
commit
0de441a5ef
|
@ -4,7 +4,7 @@ from vencoder.hubert import hubert_model
|
|||
class HubertSoft(SpeechEncoder):
|
||||
def __init__(self,vec_path = "pretrain/hubert-soft-0d54a1f4.pt",device=None):
|
||||
print("load model(s) from {}".format(vec_path))
|
||||
hubert_soft = hubert_model.hubert_soft("hubert/hubert-soft-0d54a1f4.pt")
|
||||
hubert_soft = hubert_model.hubert_soft(vec_path)
|
||||
if device is None:
|
||||
self.dev = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
||||
else:
|
||||
|
|
Loading…
Reference in New Issue