瀏覽代碼

修改prompt

jch 1 周之前
父節點
當前提交
14eba3fb92
共有 2 個文件被更改,包括 16 次插入4 次删除
  1. 8 2
      src/preprocess/generate_qw2_5_lora_sft_json.py
  2. 8 2
      src/preprocess/qw_api_url_inference.py

+ 8 - 2
src/preprocess/generate_qw2_5_lora_sft_json.py

@@ -10,8 +10,14 @@ from tqdm import tqdm
 
 image_tag = "<image>"
 
-system = "根据用户在社交平台的昵称和头像,判断用户的性别倾向;当头像不可用时,仅使用昵称进行判断;直接返回[男性|女性|未知],不要返回多余的信息。"
-prompt = "昵称:%s,头像:%s"
+system = """
+根据用户在社交平台上的姓名/昵称和头像,分析用户的性别倾向;
+先基于姓名/昵称分析,再基于头像分析;
+当头像不可用时,仅基于姓名/昵称分析;
+直接返回[男性|女性|未知],不要返回多余的信息。
+"""
+system = system.replace("\n", "")
+prompt = "姓名/昵称:%s,头像:%s"
 
 
 # LLaMA-Factory 训练模版

+ 8 - 2
src/preprocess/qw_api_url_inference.py

@@ -9,8 +9,14 @@ from tqdm import tqdm
 
 model_name_or_path = 'Qwen/Qwen2.5-VL-7B-Instruct'
 
-system = "根据用户在社交平台的昵称和头像,判断用户的性别倾向;当头像不可用时,仅使用昵称进行判断;直接返回[男性|女性|未知],不要返回多余的信息。"
-prompt = "昵称:%s, 头像:"
+system = """
+根据用户在社交平台上的姓名/昵称和头像,分析用户的性别倾向;
+先基于姓名/昵称分析,再基于头像分析;
+当头像不可用时,仅基于姓名/昵称分析;
+直接返回[男性|女性|未知],不要返回多余的信息。
+"""
+system = system.replace("\n", "")
+prompt = "姓名/昵称:%s, 头像:"
 
 
 class ModelService: