Browse Source

新增finetuning任务类型

gitlink
JeshuaRen 1 year ago
parent
commit
67b683f6fa
1 changed files with 11 additions and 0 deletions
  1. +11
    -0
      sdks/scheduler/models.go

+ 11
- 0
sdks/scheduler/models.go View File

@@ -45,6 +45,7 @@ var JobInfoTypeUnion = types.NewTypeUnion[JobInfo](
(*MultiInstanceJobInfo)(nil),
(*InstanceJobInfo)(nil),
(*UpdateMultiInstanceJobInfo)(nil),
(*FinetuningJobInfo)(nil),
)
var _ = serder.UseTypeUnionInternallyTagged(&JobInfoTypeUnion, "type")

@@ -66,6 +67,16 @@ type NormalJobInfo struct {
Services JobServicesInfo `json:"services"`
}

type FinetuningJobInfo struct {
serder.Metadata `union:"Finetuning"`
JobInfoBase
Type string `json:"type"`
Files JobFilesInfo `json:"files"`
Runtime JobRuntimeInfo `json:"runtime"`
Resources JobResourcesInfo `json:"resources"`
Services JobServicesInfo `json:"services"`
}

type DataReturnJobInfo struct {
serder.Metadata `union:"DataReturn"`
JobInfoBase


Loading…
Cancel
Save