diff --git a/browser_tests/dialog.spec.ts b/browser_tests/dialog.spec.ts index 074cea9fa..a4307dcbb 100644 --- a/browser_tests/dialog.spec.ts +++ b/browser_tests/dialog.spec.ts @@ -63,10 +63,11 @@ test.describe('Missing models warning', () => { const downloadButton = comfyPage.page.getByLabel('Download') await expect(downloadButton).toBeVisible() + const downloadPromise = comfyPage.page.waitForEvent('download') await downloadButton.click() - const downloadComplete = comfyPage.page.locator('.download-complete') - await expect(downloadComplete).toBeVisible() + const download = await downloadPromise + expect(download.suggestedFilename()).toBe('fake_model.safetensors') }) }) diff --git a/src/components/common/DeviceInfo.vue b/src/components/common/DeviceInfo.vue index 10d25b81b..8733105a0 100644 --- a/src/components/common/DeviceInfo.vue +++ b/src/components/common/DeviceInfo.vue @@ -9,7 +9,7 @@ diff --git a/src/components/common/SystemStatsPanel.vue b/src/components/common/SystemStatsPanel.vue index 2cbe2070b..df75ecb96 100644 --- a/src/components/common/SystemStatsPanel.vue +++ b/src/components/common/SystemStatsPanel.vue @@ -35,7 +35,7 @@ import TabPanel from 'primevue/tabpanel' import Divider from 'primevue/divider' import type { SystemStats } from '@/types/apiTypes' import DeviceInfo from '@/components/common/DeviceInfo.vue' -import { formatMemory } from '@/utils/formatUtil' +import { formatSize } from '@/utils/formatUtil' const props = defineProps<{ stats: SystemStats @@ -58,7 +58,7 @@ const systemColumns = [ const formatValue = (value: any, field: string) => { if (['ram_total', 'ram_free'].includes(field)) { - return formatMemory(value) + return formatSize(value) } return value } diff --git a/src/components/dialog/content/MissingModelsWarning.vue b/src/components/dialog/content/MissingModelsWarning.vue index a711fd114..a8b1b7f9c 100644 --- a/src/components/dialog/content/MissingModelsWarning.vue +++ b/src/components/dialog/content/MissingModelsWarning.vue @@ -5,67 +5,13 @@ title="Missing Models" message="When loading the graph, the following models were not found" /> - - @@ -73,12 +19,8 @@