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

pointerhacker

2024-10-15 70645e48072bf193fbf069949f1d2b10fddac8a3

 runtime/python/onnxruntime/funasr_onnx/paraformer_bin.py

@@ -322,6 +322,10 @@
            self.pred_bias = config["model_conf"]["predictor_bias"]
        else:
            self.pred_bias = 0
        if "lang" in config:
            self.language = config["lang"]
        else:
            self.language = None

    def __call__(
        self, wav_content: Union[str, np.ndarray, List[str]], hotwords: str, **kwargs