update funasr-runtime-sdk-gpu-0.1.1

雾聪

2024-07-01 31bf3a88a09f6e7c895224f4dc75e8f2c138d5c8

 funasr/models/ct_transformer/model.py

@@ -287,9 +287,7 @@
            # y, _ = self.wrapped_model(**data)
            y, _ = self.punc_forward(**data)
            _, indices = y.view(-1, y.shape[-1]).topk(1, dim=1)
            punctuations = indices
            if indices.size()[0] != 1:
                punctuations = torch.squeeze(indices)
            punctuations = torch.squeeze(indices, dim=1)
            assert punctuations.size()[0] == len(mini_sentence)

            # Search for the last Period/QuestionMark as cache