dcos

游雁

2024-07-30 d238a5ab4450aad636bbbc60d67335ca59b3bd9c

 funasr/auto/auto_model.py

@@ -114,6 +114,9 @@
        try:
            from funasr.utils.version_checker import check_for_update

            print(
                "Check update of funasr, and it would cost few times. You may disable it by set `disable_update=True` in AutoModel"
            )
            check_for_update(disable=kwargs.get("disable_update", False))
        except:
            pass
@@ -264,6 +267,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()