init commit

This commit is contained in:
Botao Chen 2024-12-10 19:39:58 -08:00
parent e5993c565e
commit dc3d9d7720
3 changed files with 68 additions and 10 deletions

View file

@ -5,7 +5,6 @@
# the root directory of this source tree.
from datetime import datetime
from typing import Optional
from llama_models.schema_utils import json_schema_type
from pydantic import BaseModel
@ -26,4 +25,4 @@ class Checkpoint(BaseModel):
epoch: int
post_training_job_id: str
path: str
training_metrics: Optional[PostTrainingMetric] = None
training_metric: PostTrainingMetric