diff --git a/funasr/models/fsmn_vad_streaming/model.py b/funasr/models/fsmn_vad_streaming/model.py index 7dfb96a05..51581ba25 100644 --- a/funasr/models/fsmn_vad_streaming/model.py +++ b/funasr/models/fsmn_vad_streaming/model.py @@ -348,7 +348,7 @@ def ComputeDecibel(self, cache: dict = {}) -> None: def ComputeScores(self, feats: torch.Tensor, cache: dict = {}) -> None: - scores = self.encoder(feats, cache=cache["encoder"]).to("cpu") # return B * T * D + scores = self.encoder(feats, cache=cache["encoder"]) # return B * T * D assert ( scores.shape[1] == feats.shape[1] ), "The shape between feats and scores does not match"