add batch_size for wss-server

雾聪

2024-03-29 d42851ad1a1da32c63caa16dd1506c1d000bb8e6

 funasr/models/e_branchformer/encoder.py

@@ -42,7 +42,7 @@
    TooShortUttError,
    check_short_utt,
)
from funasr.utils.register import register_class
from funasr.register import tables

class EBranchformerEncoderLayer(torch.nn.Module):
    """E-Branchformer encoder layer module.
@@ -174,7 +174,7 @@

        return x, mask

@register_class("encoder_classes", "EBranchformerEncoder")
@tables.register("encoder_classes", "EBranchformerEncoder")
class EBranchformerEncoder(nn.Module):
    """E-Branchformer encoder module."""