Use new definitions of Model / SKU

This commit is contained in:
Ashwin Bharambe 2024-07-31 11:36:16 -07:00
parent 156bfa0e15
commit 09cf3fe78b
8 changed files with 63 additions and 65 deletions

View file

@ -25,7 +25,7 @@ class PostTrainingSFTRequest(BaseModel):
job_uuid: str
model: PretrainedModel
model: str
dataset: TrainEvalDataset
validation_dataset: TrainEvalDataset