@daocloud-proto/hydra 0.0.0-dev-df7fe94b → 0.0.0-dev-87f727ca

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

@@ -0,0 +1,182 @@
+/* eslint-disable */
+// @ts-nocheck
+/*
+* This file is a generated Typescript file for GRPC Gateway, DO NOT MODIFY
+*/
+
+import * as HydraCommonCommon from "../../../common/common.pb"
+import * as fm from "../../../fetch.pb"
+import * as GoogleProtobufTimestamp from "../../../google/protobuf/timestamp.pb"
+
+export enum TrainingStage {
+  TRAINING_STAGE_UNSPECIFIED = "TRAINING_STAGE_UNSPECIFIED",
+  SFT = "SFT",
+  DPO = "DPO",
+}
+
+export enum FineTuningType {
+  FINE_TUNING_TYPE_UNSPECIFIED = "FINE_TUNING_TYPE_UNSPECIFIED",
+  LORA = "LORA",
+  FULL = "FULL",
+  FREEZE = "FREEZE",
+}
+
+export type ModelConfig = {
+  modelId?: string
+  modelVolumeId?: string
+  template?: string
+  trustRemoteCode?: boolean
+}
+
+export type LoraConfig = {
+  loraRank?: number
+  loraAlpha?: number
+  loraDropout?: number
+  loraTarget?: string
+  additionalTarget?: string
+  loraPlusLrRatio?: number
+  loraPlusLrEmbedding?: number
+  useRslora?: boolean
+  useDora?: boolean
+  pissaInit?: boolean
+  pissaIter?: number
+  pissaConvert?: boolean
+  createNewAdapter?: boolean
+  quantizationBit?: number
+  quantizationMethod?: string
+  ropeScaling?: string
+  booster?: string
+}
+
+export type DatasetConfig = {
+  datasetId?: string
+  datasetVolumeId?: string
+  cutoffLen?: number
+  maxSamples?: number
+  valSize?: string
+}
+
+export type TrainingConfig = {
+  numTrainEpochs?: string
+  perDeviceTrainBatchSize?: number
+  gradientAccumulationSteps?: number
+  learningRate?: string
+  lrSchedulerType?: string
+  warmupRatio?: string
+  warmupSteps?: number
+  bf16?: boolean
+  optimizer?: string
+  seed?: number
+  ddpTimeout?: number
+  maxGradNorm?: string
+  neftAlpha?: number
+  deepspeedStage?: string
+  deepspeedOffload?: boolean
+  gradientCheckpointing?: boolean
+  loggingSteps?: number
+  saveSteps?: number
+  outputSubPath?: string
+}
+
+export type DpoConfig = {
+  prefBeta?: string
+  prefLoss?: string
+  prefFtx?: string
+  dpoLabelSmoothing?: string
+  refModel?: string
+}
+
+export type ResourceQuota = {
+  request?: string
+  limit?: string
+}
+
+export type ResourceConfig = {
+  replicas?: number
+  gpuType?: string
+  gpuCount?: number
+  cpu?: ResourceQuota
+  memory?: ResourceQuota
+}
+
+export type FineTuneConfig = {
+  name?: string
+  cluster?: string
+  stage?: TrainingStage
+  finetuningType?: FineTuningType
+  model?: ModelConfig
+  lora?: LoraConfig
+  dataset?: DatasetConfig
+  training?: TrainingConfig
+  dpo?: DpoConfig
+  resources?: ResourceConfig
+  createdAt?: GoogleProtobufTimestamp.Timestamp
+  updatedAt?: GoogleProtobufTimestamp.Timestamp
+}
+
+export type CreateWSFineTuneConfigRequest = {
+  workspace?: number
+  cluster?: string
+  name?: string
+  stage?: TrainingStage
+  finetuningType?: FineTuningType
+  model?: ModelConfig
+  lora?: LoraConfig
+  dataset?: DatasetConfig
+  training?: TrainingConfig
+  dpo?: DpoConfig
+  resources?: ResourceConfig
+}
+
+export type UpdateWSFineTuneConfigRequest = {
+  workspace?: number
+  cluster?: string
+  name?: string
+  stage?: TrainingStage
+  finetuningType?: FineTuningType
+  model?: ModelConfig
+  lora?: LoraConfig
+  dataset?: DatasetConfig
+  training?: TrainingConfig
+  dpo?: DpoConfig
+  resources?: ResourceConfig
+}
+
+export type GetWSFineTuneConfigRequest = {
+  workspace?: number
+  name?: string
+}
+
+export type ListWSFineTuneConfigsRequest = {
+  workspace?: number
+  cluster?: string
+  page?: HydraCommonCommon.Pagination
+}
+
+export type ListWSFineTuneConfigsResponse = {
+  items?: FineTuneConfig[]
+  page?: HydraCommonCommon.Pagination
+}
+
+export type DeleteWSFineTuneConfigRequest = {
+  workspace?: number
+  name?: string
+}
+
+export class WSFineTuneConfigManagement {
+  static CreateWSFineTuneConfig(req: CreateWSFineTuneConfigRequest, initReq?: fm.InitReq): Promise<FineTuneConfig> {
+    return fm.fetchReq<CreateWSFineTuneConfigRequest, FineTuneConfig>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/train-configs`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+  static UpdateWSFineTuneConfig(req: UpdateWSFineTuneConfigRequest, initReq?: fm.InitReq): Promise<FineTuneConfig> {
+    return fm.fetchReq<UpdateWSFineTuneConfigRequest, FineTuneConfig>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/train-configs/${req["name"]}`, {...initReq, method: "PUT", body: JSON.stringify(req, fm.replacer)})
+  }
+  static GetWSFineTuneConfig(req: GetWSFineTuneConfigRequest, initReq?: fm.InitReq): Promise<FineTuneConfig> {
+    return fm.fetchReq<GetWSFineTuneConfigRequest, FineTuneConfig>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/train-configs/${req["name"]}?${fm.renderURLSearchParams(req, ["workspace", "name"])}`, {...initReq, method: "GET"})
+  }
+  static ListWSFineTuneConfigs(req: ListWSFineTuneConfigsRequest, initReq?: fm.InitReq): Promise<ListWSFineTuneConfigsResponse> {
+    return fm.fetchReq<ListWSFineTuneConfigsRequest, ListWSFineTuneConfigsResponse>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/train-configs?${fm.renderURLSearchParams(req, ["workspace"])}`, {...initReq, method: "GET"})
+  }
+  static DeleteWSFineTuneConfig(req: DeleteWSFineTuneConfigRequest, initReq?: fm.InitReq): Promise<FineTuneConfig> {
+    return fm.fetchReq<DeleteWSFineTuneConfigRequest, FineTuneConfig>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/train-configs/${req["name"]}`, {...initReq, method: "DELETE"})
+  }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@daocloud-proto/hydra",
-  "version": "v0.0.0-dev-df7fe94b",
+  "version": "v0.0.0-dev-87f727ca",
   "description": "",
   "author": "",
   "license": "ISC"