Add default case for training type as full

togethercomputer · Aug 28, 2024 · bdea27f · bdea27f
1 parent fe40c59
commit bdea27f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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())