| | |
|---|
| | | kwargs["init_param"] = init_param |
|---|
| | | 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")): |
|---|
| | | kwargs["tokenizer_conf"]["token_list"] = os.path.join(model_or_path, "tokens.json") |
|---|
| | | if os.path.exists(os.path.join(model_or_path, "seg_dict")): |
|---|
| | | kwargs["tokenizer_conf"]["seg_dict"] = os.path.join(model_or_path, "seg_dict") |
|---|
| | | if os.path.exists(os.path.join(model_or_path, "bpe.model")): |
|---|
| | | kwargs["tokenizer_conf"]["bpemodel"] = os.path.join(model_or_path, "bpe.model") |
|---|
| | | kwargs["model"] = cfg["model"] |
|---|
| | | kwargs["frontend_conf"]["cmvn_file"] = os.path.join(model_or_path, "am.mvn") |
|---|
| | | if os.path.exists(os.path.join(model_or_path, "am.mvn")): |
|---|
| | | kwargs["frontend_conf"]["cmvn_file"] = os.path.join(model_or_path, "am.mvn") |
|---|
| | | |
|---|
| | | return OmegaConf.to_container(kwargs, resolve=True) |
|---|
| | | |
|---|
