From d238a5ab4450aad636bbbc60d67335ca59b3bd9c Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 30 七月 2024 17:45:13 +0800
Subject: [PATCH] dcos
---
funasr/auto/auto_model.py | 5 +++++
1 files changed, 5 insertions(+), 0 deletions(-)
diff --git a/funasr/auto/auto_model.py b/funasr/auto/auto_model.py
index acd4c72..f735f18 100644
--- a/funasr/auto/auto_model.py
+++ b/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()
--
Gitblit v1.9.1