feat: 对话补全训练数据生成

2025-05-13 06:42:43 +00:00 · 2025-05-13 06:42:43 +00:00 · c2d49e885e
commit c2d49e885e
parent 89f10d841d
1 changed files with 2 additions and 1 deletions
--- a/data_generate/query_completion/select_final_data.py
+++ b/data_generate/query_completion/select_final_data.py
@ -8,7 +8,8 @@

 import json
 import pandas as pd
-
+import torch
+import torch.nn.functional as F

 def load_excel_column(file_path, column_index=2):
    """读取 Excel 文件的第三列，并将每一行解析为 dict"""