update

modelscope · Dec 26, 2024 · d2c88d1 · d2c88d1
1 parent 9b75b48
commit d2c88d1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/swift/llm/template/base.py b/swift/llm/template/base.py
@@ -181,7 +181,8 @@ def _kto_encode(self, inputs: StdTemplateInputs) -> Dict[str, Any]:
 
     def _seq_cls_encode(self, inputs: StdTemplateInputs) -> Dict[str, Any]:
         encoded = self._encode(inputs)
-        encoded['labels'] = int(inputs.label)
+        if inputs.label is not None:
+            encoded['labels'] = int(inputs.label)
         return encoded
 
     def encode(self,