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), (*MultiInstanceJobInfo)(nil),
(*InstanceJobInfo)(nil), (*InstanceJobInfo)(nil),
(*UpdateMultiInstanceJobInfo)(nil), (*UpdateMultiInstanceJobInfo)(nil),
(*FinetuningJobInfo)(nil),
) )
var _ = serder.UseTypeUnionInternallyTagged(&JobInfoTypeUnion, "type") var _ = serder.UseTypeUnionInternallyTagged(&JobInfoTypeUnion, "type")


@@ -66,6 +67,16 @@ type NormalJobInfo struct {
Services JobServicesInfo `json:"services"` 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 { type DataReturnJobInfo struct {
serder.Metadata `union:"DataReturn"` serder.Metadata `union:"DataReturn"`
JobInfoBase JobInfoBase


Loading…
Cancel
Save