diff --git a/pyproject.toml b/pyproject.toml index 0acd08e..f06c6d5 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -12,7 +12,7 @@ build-backend = "poetry.masonry.api" [tool.poetry] name = "together" -version = "1.5.32" +version = "1.5.33" authors = ["Together AI "] description = "Python client for Together's Cloud Platform! Note: SDK 2.0 is now available at https://github.com/togethercomputer/together-py" readme = "README.md" diff --git a/src/together/types/finetune.py b/src/together/types/finetune.py index 37250ae..0eb7402 100644 --- a/src/together/types/finetune.py +++ b/src/together/types/finetune.py @@ -337,8 +337,8 @@ class FinetunePriceEstimationRequest(BaseModel): model: str n_epochs: int n_evals: int - training_type: TrainingType - training_method: TrainingMethod + training_type: LoRATrainingType | FullTrainingType + training_method: TrainingMethodSFT | TrainingMethodDPO class FinetunePriceEstimationResponse(BaseModel):