Merge pull request #2206 from yijinsheng/triton_gpu

Yuekai Zhang

2024-11-12 ef1d7b3f12535d73bf9046085d14d192c54cf7da

 funasr/models/whisper/model.py

@@ -28,6 +28,7 @@
@tables.register("model_classes", "Whisper-large-v1")
@tables.register("model_classes", "Whisper-large-v2")
@tables.register("model_classes", "Whisper-large-v3")
@tables.register("model_classes", "Whisper-large-v3-turbo")
@tables.register("model_classes", "WhisperWarp")
class WhisperWarp(nn.Module):
    def __init__(self, *args, **kwargs):