From 47209220c653e2faa2c6c83801874b9f1df7d7dc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= Date: Fri, 14 Jun 2024 09:49:22 +0800 Subject: [PATCH] decoding --- funasr/download/download_from_hub.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/funasr/download/download_from_hub.py b/funasr/download/download_from_hub.py index d3c164464..46263c93a 100644 --- a/funasr/download/download_from_hub.py +++ b/funasr/download/download_from_hub.py @@ -56,13 +56,13 @@ def download_from_ms(**kwargs): config = OmegaConf.load(cfg["config"]) kwargs = OmegaConf.merge(config, cfg) kwargs["model"] = config["model"] - elif os.path.exists(os.path.join(model_or_path, "config.yaml")) and os.path.exists( - os.path.join(model_or_path, "model.pt") - ): + elif os.path.exists(os.path.join(model_or_path, "config.yaml")): config = OmegaConf.load(os.path.join(model_or_path, "config.yaml")) kwargs = OmegaConf.merge(config, kwargs) init_param = os.path.join(model_or_path, "model.pt") - kwargs["init_param"] = init_param + if "init_param" not in kwargs or not os.path.exists(kwargs["init_param"]): + kwargs["init_param"] = init_param + assert os.path.exists(kwargs["init_param"]), "init_param does not exist" if os.path.exists(os.path.join(model_or_path, "tokens.txt")): kwargs["tokenizer_conf"]["token_list"] = os.path.join(model_or_path, "tokens.txt") if os.path.exists(os.path.join(model_or_path, "tokens.json")):