游雁
2024-07-27 69370ed3d62a06989860a77989c1651e90b87466
funasr/auto/auto_model.py
@@ -114,7 +114,7 @@
        try:
            from funasr.utils.version_checker import check_for_update
            check_for_update()
            check_for_update(disable=kwargs.get("disable_update", False))
        except:
            pass
@@ -264,6 +264,8 @@
    def inference(self, input, input_len=None, model=None, kwargs=None, key=None, **cfg):
        kwargs = self.kwargs if kwargs is None else kwargs
        if "cache" in kwargs:
            kwargs.pop("cache")
        deep_update(kwargs, cfg)
        model = self.model if model is None else model
        model.eval()
@@ -337,7 +339,7 @@
        end_vad = time.time()
        #  FIX(gcf): concat the vad clips for sense vocie model for better aed
        if kwargs.get("merge_vad", False):
        if cfg.get("merge_vad", False):
            for i in range(len(res)):
                res[i]["value"] = merge_vad(
                    res[i]["value"], kwargs.get("merge_length_s", 15) * 1000