feat: add gpu options

This commit is contained in:
BennyKok
2024-01-11 21:57:30 +08:00
parent 6e3b7f0435
commit 29bf7636b1
10 changed files with 792 additions and 5 deletions
+3
View File
@@ -104,6 +104,8 @@ export const workflowRunOrigin = pgEnum("workflow_run_origin", [
"api",
]);
export const machineGPUOptions = pgEnum("machine_gpu", ["T4", "A10G", "A100"]);
export const machinesType = pgEnum("machine_type", [
"classic",
"runpod-serverless",
@@ -205,6 +207,7 @@ export const machinesTable = dbSchema.table("machines", {
status: machinesStatus("status").notNull().default("ready"),
snapshot: jsonb("snapshot").$type<any>(),
models: jsonb("models").$type<any>(),
gpu: machineGPUOptions("gpu"),
build_machine_instance_id: text("build_machine_instance_id"),
build_log: text("build_log"),
});