From e8a9e8e28c439d3579df88b00f088ad3eb752f61 Mon Sep 17 00:00:00 2001 From: Louis Date: Tue, 22 Oct 2024 16:16:43 +0700 Subject: [PATCH] fix: inherits model decoration metadata from Jan --- web/hooks/useModels.ts | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/web/hooks/useModels.ts b/web/hooks/useModels.ts index 742d09beb..f884a591e 100644 --- a/web/hooks/useModels.ts +++ b/web/hooks/useModels.ts @@ -30,10 +30,17 @@ const useModels = () => { const getData = useCallback(() => { const getDownloadedModels = async () => { - const localModels = await getModels() - const hubModels = ModelManager.instance().models.values().toArray() + const localModels = (await getModels()).map((e) => ({ + ...e, + name: ModelManager.instance().models.get(e.id)?.name ?? e.name, + metadata: + ModelManager.instance().models.get(e.id)?.metadata ?? e.metadata, + })) - const remoteModels = hubModels.filter((e) => !isLocalEngine(e.engine)) + const remoteModels = ModelManager.instance() + .models.values() + .toArray() + .filter((e) => !isLocalEngine(e.engine)) setDownloadedModels([...localModels, ...remoteModels]) }