in src/autotrain/app/api_routes.py [0:0]
def validate_params(cls, values):
if values.get("task") == "llm:sft":
values["params"] = LLMSFTTrainingParamsAPI(**values["params"])
elif values.get("task") == "llm:dpo":
values["params"] = LLMDPOTrainingParamsAPI(**values["params"])
elif values.get("task") == "llm:orpo":
values["params"] = LLMORPOTrainingParamsAPI(**values["params"])
elif values.get("task") == "llm:generic":
values["params"] = LLMGenericTrainingParamsAPI(**values["params"])
elif values.get("task") == "llm:reward":
values["params"] = LLMRewardTrainingParamsAPI(**values["params"])
elif values.get("task") == "seq2seq":
values["params"] = Seq2SeqParamsAPI(**values["params"])
elif values.get("task") == "image-classification":
values["params"] = ImageClassificationParamsAPI(**values["params"])
elif values.get("task") == "tabular-classification":
values["params"] = TabularClassificationParamsAPI(**values["params"])
elif values.get("task") == "tabular-regression":
values["params"] = TabularRegressionParamsAPI(**values["params"])
elif values.get("task") == "text-classification":
values["params"] = TextClassificationParamsAPI(**values["params"])
elif values.get("task") == "text-regression":
values["params"] = TextRegressionParamsAPI(**values["params"])
elif values.get("task") == "token-classification":
values["params"] = TokenClassificationParamsAPI(**values["params"])
elif values.get("task").startswith("st:"):
values["params"] = SentenceTransformersParamsAPI(**values["params"])
elif values.get("task") == "image-regression":
values["params"] = ImageRegressionParamsAPI(**values["params"])
elif values.get("task").startswith("vlm:"):
values["params"] = VLMTrainingParamsAPI(**values["params"])
elif values.get("task") == "extractive-question-answering":
values["params"] = ExtractiveQuestionAnsweringParamsAPI(**values["params"])
elif values.get("task") == "image-object-detection":
values["params"] = ObjectDetectionParamsAPI(**values["params"])
return values