Merge pull request #85 from togethercomputer/cmanuel/eng-24979-move-train_on_inputs-to-the-parameters-of-the-sft-training

Move train_on_inputs to SFTType

Author: connermanuel

openapi.yaml

@@ -729,6 +729,7 @@ paths:
                   type: boolean
                   default: auto
                   description: Whether to mask the user messages in conversational data or prompts in instruction data.
+                  deprecated: true
                 training_method:
                   type: object
                   oneOf:
@@ -2879,15 +2880,25 @@ components:
         - type
         - lora_r
         - lora_alpha
+
     TrainingMethodSFT:
       type: object
       properties:
         method:
           type: string
           enum: ["sft"]
+        train_on_inputs:
+          oneOf:
+            - type: boolean
+            - type: string
+              enum:
+                - auto
+          type: boolean
+          default: auto
+          description: Whether to mask the user messages in conversational data or prompts in instruction data.
       required:
         - method
-
+        - train_on_inputs
     TrainingMethodDPO:
       type: object
       properties: