diff --git a/src/components/common/DeviceInfo.vue b/src/components/common/DeviceInfo.vue index d89e8b9c6..10d25b81b 100644 --- a/src/components/common/DeviceInfo.vue +++ b/src/components/common/DeviceInfo.vue @@ -9,6 +9,7 @@ diff --git a/src/types/apiTypes.ts b/src/types/apiTypes.ts index 20c4a4ed6..ea581b59e 100644 --- a/src/types/apiTypes.ts +++ b/src/types/apiTypes.ts @@ -410,7 +410,9 @@ export const zSystemStats = z.object({ embedded_python: z.boolean(), comfyui_version: z.string(), pytorch_version: z.string(), - argv: z.array(z.string()) + argv: z.array(z.string()), + ram_total: z.number(), + ram_free: z.number() }), devices: z.array(zDeviceStats) }) diff --git a/src/utils/formatUtil.ts b/src/utils/formatUtil.ts index 2d702b769..bbddf8f44 100644 --- a/src/utils/formatUtil.ts +++ b/src/utils/formatUtil.ts @@ -59,3 +59,15 @@ export function formatNumberWithSuffix( return `${formattedNum}${suffixes[exp]}` } + +export function formatMemory(value?: number) { + if (value === null || value === undefined) { + return '-' + } + + const mb = Math.round(value / (1024 * 1024)) + if (mb >= 1024) { + return `${(mb / 1024).toFixed(2)} GB` + } + return `${mb} MB` +}