mirror of
https://github.com/modelscope/FunASR.git
synced 2026-10-12 02:19:18 +00:00
download configuration.json
This commit is contained in:
@@ -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 = {}
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user