use integrated hfl/chinese-roberta-wwm-ext-large

text/chinese_bert.py

@@ -1,6 +1,5 @@
 import sys
 import torch
-from huggingface_hub import snapshot_download
 from transformers import AutoTokenizer, AutoModelForMaskedLM
 
 device = torch.device(
@@ -14,7 +13,7 @@ device = torch.device(
 )
 
 # 模型下载
-model_dir = snapshot_download("hfl/chinese-roberta-wwm-ext-large")
+model_dir = "hfl/chinese-roberta-wwm-ext-large"
 tokenizer = AutoTokenizer.from_pretrained(model_dir)
 model = AutoModelForMaskedLM.from_pretrained(model_dir).to(device)