From a1e98765230de5f775a9b77fa54a4f596e6ef522 Mon Sep 17 00:00:00 2001
From: Jintao Huang <huangjintao.hjt@alibaba-inc.com>
Date: Tue, 24 Dec 2024 22:18:18 +0800
Subject: [PATCH] fix

---
 examples/train/plugins/train_cls.sh | 1 +
 1 file changed, 1 insertion(+)

diff --git a/examples/train/plugins/train_cls.sh b/examples/train/plugins/train_cls.sh
index 8b9aa92c2..48f45eaa1 100644
--- a/examples/train/plugins/train_cls.sh
+++ b/examples/train/plugins/train_cls.sh
@@ -8,6 +8,7 @@ swift sft \
     --dataset simpleai/HC3-Chinese:baike_cls#1000 \
     --num_train_epochs 1 \
     --num_labels 2 \
+    --task_type seq_cls \
     --per_device_train_batch_size 1 \
     --learning_rate 1e-5 \
     --gradient_accumulation_steps 16 \