@daocloud-proto/hydra 0.18.0-dev-17 → 0.18.0-dev-18
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.
|
@@ -8,15 +8,6 @@ import * as HydraCommonCommon from "../../../common/common.pb"
|
|
|
8
8
|
import * as fm from "../../../fetch.pb"
|
|
9
9
|
import * as GoogleProtobufTimestamp from "../../../google/protobuf/timestamp.pb"
|
|
10
10
|
|
|
11
|
-
type Absent<T, K extends keyof T> = { [k in Exclude<keyof T, K>]?: undefined };
|
|
12
|
-
type OneOf<T> =
|
|
13
|
-
| { [k in keyof T]?: undefined }
|
|
14
|
-
| (
|
|
15
|
-
keyof T extends infer K ?
|
|
16
|
-
(K extends string & keyof T ? { [k in K]: T[K] } & Absent<T, K>
|
|
17
|
-
: never)
|
|
18
|
-
: never);
|
|
19
|
-
|
|
20
11
|
export enum TrainingStage {
|
|
21
12
|
TRAINING_STAGE_UNSPECIFIED = "TRAINING_STAGE_UNSPECIFIED",
|
|
22
13
|
SFT = "SFT",
|
|
@@ -138,19 +129,13 @@ export type DpoConfig = {
|
|
|
138
129
|
refModel?: string
|
|
139
130
|
}
|
|
140
131
|
|
|
141
|
-
|
|
142
|
-
/* hydra modified */ export type BaseResourceConfig = {
|
|
132
|
+
export type ResourceConfig = {
|
|
143
133
|
gpuType?: string
|
|
144
|
-
gpuCount?: number
|
|
145
134
|
cpu?: number
|
|
146
135
|
memory?: number
|
|
147
|
-
perGpuMemory?: number
|
|
148
136
|
gpuResourceValues?: {[key: string]: string}
|
|
149
137
|
}
|
|
150
138
|
|
|
151
|
-
export type ResourceConfig = BaseResourceConfig
|
|
152
|
-
& OneOf<{ gpuCores: number }>
|
|
153
|
-
|
|
154
139
|
export type FineTuneConfig = {
|
|
155
140
|
name?: string
|
|
156
141
|
cluster?: string
|
|
@@ -145,13 +145,10 @@ export type ConfigMapMountConfig = {
|
|
|
145
145
|
cpu?: number
|
|
146
146
|
memory?: number
|
|
147
147
|
gpuType?: string
|
|
148
|
-
gpuCount?: number
|
|
149
|
-
perGpuMemory?: number
|
|
150
148
|
gpuResourceValues?: {[key: string]: string}
|
|
151
149
|
}
|
|
152
150
|
|
|
153
151
|
export type ResourceConfig = BaseResourceConfig
|
|
154
|
-
& OneOf<{ gpuCores: number }>
|
|
155
152
|
& OneOf<{ shmSize: number }>
|
|
156
153
|
& OneOf<{ enableShm: boolean }>
|
|
157
154
|
|