diff --git a/src/together/cli/api/finetune.py b/src/together/cli/api/finetune.py index bd92606c..8b4bf135 100644 --- a/src/together/cli/api/finetune.py +++ b/src/together/cli/api/finetune.py @@ -1,3 +1,5 @@ +from __future__ import annotations + import json from textwrap import wrap diff --git a/src/together/types/finetune.py b/src/together/types/finetune.py index bd8eef21..26d2f2ab 100644 --- a/src/together/types/finetune.py +++ b/src/together/types/finetune.py @@ -230,7 +230,7 @@ class FinetuneResponse(BaseModel): @field_validator("training_type") @classmethod def validate_training_type(cls, v: TrainingType) -> TrainingType: - if v.type == "Full": + if v.type == "Full" or v.type == "": return FullTrainingType(**v.model_dump()) elif v.type == "Lora": return LoRATrainingType(**v.model_dump())