重构模型训练

2025-03-09 14:50:42 +08:00 · 2025-03-09 14:50:42 +08:00 · 95843aae79
parent aabc01d209
commit 95843aae79
1 changed files with 2 additions and 2 deletions
--- a/uie/test_model.py
+++ b/uie/test_model.py
@ -7,7 +7,7 @@ model = ErnieForTokenClassification.from_pretrained(model_path)
 tokenizer = ErnieTokenizer.from_pretrained(model_path)

 # 2. 处理输入文本
-text = "能否帮我进入日计划应用"
+text = "今天杨柳220kV变电站220kV南坪间隔扩建工程有多少作业计划"
 inputs = tokenizer(text, max_len=512, return_tensors="pd")

 # 3. 进行预测