19 lines
349 B
Python
19 lines
349 B
Python
|
from pydantic import BaseModel
|
||
|
|
|
||
|
|
|
||
|
|
class FinetuneStartRequest(BaseModel):
|
||
|
|
jsonl_url: str
|
||
|
|
base_model: str
|
||
|
|
hyperparams: dict | None = None
|
||
|
|
|
||
|
|
|
||
|
|
class FinetuneStartResponse(BaseModel):
|
||
|
|
job_id: str
|
||
|
|
|
||
|
|
|
||
|
|
class FinetuneStatusResponse(BaseModel):
|
||
|
|
job_id: str
|
||
|
|
status: str
|
||
|
|
progress: int | None = None
|
||
|
|
error_message: str | None = None
|