数据并行可能导致的模型训练报错 (#2139)

pointerhacker

2024-10-15 70645e48072bf193fbf069949f1d2b10fddac8a3

 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):