update device

shixian.shi

2024-01-12 c3442d9566f5a2011c95b0d2998958a1b5348564

 funasr/models/ct_transformer/encoder.py

@@ -31,7 +31,7 @@

from funasr.models.ctc.ctc import CTC

from funasr.utils.register import register_class
from funasr.register import tables

class EncoderLayerSANM(nn.Module):
    def __init__(
@@ -155,7 +155,7 @@
        return x, cache


@register_class("encoder_classes", "SANMVadEncoder")
@tables.register("encoder_classes", "SANMVadEncoder")
class SANMVadEncoder(nn.Module):
    """
    Author: Speech Lab of DAMO Academy, Alibaba Group