Update HubertSoft.py

2023-05-14 23:50:08 +08:00 · 2023-05-14 23:50:08 +08:00 · ee9e6e7994
parent 090492b8c3
commit ee9e6e7994
1 changed files with 0 additions and 1 deletions
--- a/vencoder/HubertSoft.py
+++ b/vencoder/HubertSoft.py
@ -17,7 +17,6 @@ class HubertSoft(SpeechEncoder):
        if feats.dim() == 2:  # double channels
          feats = feats.mean(-1)
        assert feats.dim() == 1, feats.dim()
-        feats = feats.view(1, -1)
        with torch.inference_mode():
          units = self.model.units(feats)
          return units.transpose(1,2)