Fix bert path retrieval issue when initializing G2PWPinyin

2025-10-05 22:20:01 +08:00 · 2025-04-06 22:27:03 +08:00 · 2025-04-06 22:27:03 +08:00 · d8adfe5c90
commit d8adfe5c90
parent 28effc2d44
1 changed files with 1 additions and 0 deletions
--- a/api.py
+++ b/api.py
@ -1007,6 +1007,7 @@ else:
    logger.info(f"数据类型: int16")

 # 初始化模型
+os.environ["bert_path"] = bert_path
 cnhubert.cnhubert_base_path = cnhubert_base_path
 tokenizer = AutoTokenizer.from_pretrained(bert_path)
 bert_model = AutoModelForMaskedLM.from_pretrained(bert_path)