@daocloud-proto/hydra 0.0.0-dev-cb927b7b → 0.0.0-dev-d8ebaf68

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -89,6 +89,7 @@ export type JWTPlugin = {
89
89
  enabled?: boolean
90
90
  action?: PluginAction
91
91
  headerKey?: string
92
+ required?: boolean
92
93
  }
93
94
 
94
95
  export type ModelWhitelistPlugin = {
@@ -102,25 +102,26 @@ export type DatasetConfig = {
102
102
  datasetVolumeId?: string
103
103
  cutoffLen?: number
104
104
  maxSamples?: number
105
- valSize?: string
105
+ valSize?: number
106
106
  datasetName?: string
107
+ datasetTag?: string
107
108
  }
108
109
 
109
110
  export type TrainingConfig = {
110
- numTrainEpochs?: string
111
+ numTrainEpochs?: number
111
112
  perDeviceTrainBatchSize?: number
112
113
  gradientAccumulationSteps?: number
113
- learningRate?: string
114
+ learningRate?: number
114
115
  lrSchedulerType?: LrSchedulerType
115
- warmupRatio?: string
116
+ warmupRatio?: number
116
117
  warmupSteps?: number
117
118
  computeType?: ComputeType
118
119
  optimizer?: OptimizerType
119
120
  seed?: number
120
121
  ddpTimeout?: number
121
- maxGradNorm?: string
122
+ maxGradNorm?: number
122
123
  neftAlpha?: number
123
- deepspeedStage?: string
124
+ deepspeedStage?: number
124
125
  deepspeedOffload?: boolean
125
126
  gradientCheckpointing?: boolean
126
127
  loggingSteps?: number
@@ -129,10 +130,10 @@ export type TrainingConfig = {
129
130
  }
130
131
 
131
132
  export type DpoConfig = {
132
- prefBeta?: string
133
+ prefBeta?: number
133
134
  prefLoss?: string
134
- prefFtx?: string
135
- dpoLabelSmoothing?: string
135
+ prefFtx?: number
136
+ dpoLabelSmoothing?: number
136
137
  refModel?: string
137
138
  }
138
139
 
@@ -163,6 +164,7 @@ export type FineTuneConfig = {
163
164
  createdAt?: GoogleProtobufTimestamp.Timestamp
164
165
  updatedAt?: GoogleProtobufTimestamp.Timestamp
165
166
  modelSource?: boolean
167
+ workspace?: number
166
168
  }
167
169
 
168
170
  export type CreateWSFineTuneConfigRequest = {
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@daocloud-proto/hydra",
3
- "version": "v0.0.0-dev-cb927b7b",
3
+ "version": "v0.0.0-dev-d8ebaf68",
4
4
  "description": "",
5
5
  "author": "",
6
6
  "license": "ISC"