diff --git a/src/components/sidebar/tabs/ModelLibrarySidebarTab.vue b/src/components/sidebar/tabs/ModelLibrarySidebarTab.vue index a325a9921..86e777012 100644 --- a/src/components/sidebar/tabs/ModelLibrarySidebarTab.vue +++ b/src/components/sidebar/tabs/ModelLibrarySidebarTab.vue @@ -75,7 +75,7 @@ const { expandNode, toggleNodeOnEvent } = useTreeExpansion(expandedKeys) const filteredModels = ref([]) const handleSearch = async (query: string) => { if (!query) { - filteredModels.value = modelStore.models + filteredModels.value = [] expandedKeys.value = {} return } @@ -96,7 +96,7 @@ type ModelOrFolder = ComfyModelDef | ModelFolder const root = computed(() => { const allNodes: ModelOrFolder[] = [ ...modelStore.modelFolders, - ...filteredModels.value + ...(searchQuery.value ? filteredModels.value : modelStore.models) ] return buildTree(allNodes, (modelOrFolder: ModelOrFolder) => modelOrFolder.key.split('/')