@daocloud-proto/baize 0.123.0 → 0.124.0

package/management-api/job/v1alpha1/common.pb.ts

@@ -12,6 +12,12 @@ export enum QueueType {
   KUEUE = "KUEUE",
 }
 
+export enum PlacementPolicy {
+  PLACEMENT_POLICY_UNSPECIFIED = "PLACEMENT_POLICY_UNSPECIFIED",
+  BALANCED = "BALANCED",
+  GROUPED = "GROUPED",
+}
+
 export enum TrainingMode {
   TRAINING_MODE_UNSPECIFIED = "TRAINING_MODE_UNSPECIFIED",
   SINGLE = "SINGLE",
@@ -63,6 +69,7 @@ export type JobCreationBaseConfig = {
   noOverrideEnvPath?: boolean
   imageConfig?: BaizeManagement_apiImageV1alpha1Image.ImageConfig
   preflight?: boolean
+  placement?: PlacementPolicy
 }
 
 export type JobRoleDifferenceConfig = {

package/management-api/job/v1alpha1/job.pb.ts

@@ -142,7 +142,7 @@ export type CleanCheckpointConfig = {
   dirs?: string[]
   retained?: number
   timezone?: string
-  prunePattern?: string
+  whitelist?: string
 }
 
 export type TrainingConfig = {

package/management-api/serving/v1alpha1/serving.pb.ts

@@ -156,6 +156,7 @@ export type InferenceServing = {
   hpaConfig?: HPAConfig
   image?: string
   imageConfig?: BaizeManagement_apiImageV1alpha1Image.ImageConfig
+  targetGpus?: string[]
 }
 
 export type InferenceServingStatusModelStatus = {
@@ -215,6 +216,7 @@ export type CreateInferenceServingRequest = {
   workspace?: number
   image?: string
   imageConfig?: BaizeManagement_apiImageV1alpha1Image.ImageConfig
+  targetGpus?: string[]
 }
 
 export type UpdateInferenceServingRequest = {
@@ -232,6 +234,7 @@ export type UpdateInferenceServingRequest = {
   workspace?: number
   image?: string
   imageConfig?: BaizeManagement_apiImageV1alpha1Image.ImageConfig
+  targetGpus?: string[]
 }
 
 export type SingleInferenceServingRequest = {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@daocloud-proto/baize",
-  "version": "v0.123.0",
+  "version": "v0.124.0",
   "description": "",
   "author": "",
   "license": "ISC"