remove load wiehgt false part

This commit is contained in:
samiabat 2025-03-25 00:44:09 +03:00
parent 9e96a70762
commit 944119b720

4
api.py
View File

@ -998,9 +998,9 @@ else:
# 初始化模型 # 初始化模型
cnhubert.cnhubert_base_path = cnhubert_base_path cnhubert.cnhubert_base_path = cnhubert_base_path
# tokenizer = AutoTokenizer.from_pretrained(bert_path) # tokenizer = AutoTokenizer.from_pretrained(bert_path)
tokenizer = AutoTokenizer.from_pretrained(bert_path, weights_only=False) tokenizer = AutoTokenizer.from_pretrained(bert_path)
bert_model = AutoModelForMaskedLM.from_pretrained(bert_path, weights_only=False) bert_model = AutoModelForMaskedLM.from_pretrained(bert_path)
ssl_model = cnhubert.get_model() ssl_model = cnhubert.get_model()
if is_half: if is_half:
bert_model = bert_model.half().to(device) bert_model = bert_model.half().to(device)