download configuration.json

This commit is contained in:
游雁
2024-01-15 11:07:33 +08:00
parent 8912e0696a
commit 831c48a886
2 changed files with 6 additions and 7 deletions
+2 -2
View File
@@ -22,6 +22,7 @@ def download_from_ms(**kwargs):
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 @@ def download_from_ms(**kwargs):
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 = {}
+4 -5
View File
@@ -1,14 +1,13 @@
#!/usr/bin/env python3
# -*- encoding: utf-8 -*-
# Copyright FunASR (https://github.com/alibaba-damo-academy/FunASR). All Rights Reserved.
# MIT License (https://opensource.org/licenses/MIT)
import logging
from typing import Dict
from typing import List
from typing import Optional
from typing import Tuple
from typing import Union
import tempfile
import codecs
import requests
import re
import copy
import torch
import torch.nn as nn