diff --git a/web/hooks/useModels.ts b/web/hooks/useModels.ts index 75c86035a..309cc4782 100644 --- a/web/hooks/useModels.ts +++ b/web/hooks/useModels.ts @@ -89,13 +89,13 @@ const useModels = () => { const cachedModels = ModelManager.instance().models.values().toArray() const toUpdate = [ ...downloadedModels, - ...cachedModels.filter( - (e: Model) => !downloadedModels.some((g: Model) => g.id === e.id) + ...cachedModels.filter((e) => !isLocalEngine(e.engine) && !downloadedModels.some((g: Model) => g.id === e.id) ), ] setDownloadedModels(toUpdate) - }, [downloadedModels, setDownloadedModels]) + setExtensionModels(cachedModels) + }, [downloadedModels, setDownloadedModels, setExtensionModels]) const getModels = async (): Promise => extensionManager