shixian.shi
2024-01-15 55c09aeaa25b4bb88a50e09ba68fa6ff00a6d676
funasr/download/download_from_hub.py
@@ -22,6 +22,7 @@
   
   config = os.path.join(model_or_path, "config.yaml")
   if os.path.exists(config) and os.path.exists(os.path.join(model_or_path, "model.pb")):
      config = OmegaConf.load(config)
      kwargs = OmegaConf.merge(config, kwargs)
      init_param = os.path.join(model_or_path, "model.pb")
@@ -39,8 +40,7 @@
         kwargs["frontend_conf"]["cmvn_file"] = os.path.join(model_or_path, "am.mvn")
      if os.path.exists(os.path.join(model_or_path, "jieba_usr_dict")):
         kwargs["jieba_usr_dict"] = os.path.join(model_or_path, "jieba_usr_dict")
   else:# configuration.json
      assert os.path.exists(os.path.join(model_or_path, "configuration.json"))
   elif os.path.exists(os.path.join(model_or_path, "configuration.json")):
      with open(os.path.join(model_or_path, "configuration.json"), 'r', encoding='utf-8') as f:
         conf_json = json.load(f)
         cfg = {}