Remove question mark badge on folders in model library tree

src/components/common/TreeExplorerTreeNode.vue

@@ -21,8 +21,8 @@
         <slot name="after-label" :node="props.node"></slot>
       </span>
       <Badge
-        v-if="!props.node.leaf"
-        :value="props.node.badgeText ?? props.node.totalLeaves"
+        v-if="showNodeBadgeText"
+        :value="nodeBadgeText"
         severity="secondary"
         class="leaf-count-badge"
       />
@@ -62,6 +62,17 @@ const emit = defineEmits<{
   (e: 'dragEnd', node: RenderedTreeExplorerNode): void
 }>()
 
+const nodeBadgeText = computed<string>(() => {
+  if (props.node.leaf) {
+    return ''
+  }
+  if (props.node.badgeText !== undefined && props.node.badgeText !== null) {
+    return props.node.badgeText
+  }
+  return props.node.totalLeaves.toString()
+})
+const showNodeBadgeText = computed<boolean>(() => nodeBadgeText.value !== '')
+
 const labelEditable = computed<boolean>(() => !!props.node.handleRename)
 const renameEditingNode =
   inject<Ref<TreeExplorerNode | null>>('renameEditingNode')

src/components/sidebar/tabs/ModelLibrarySidebarTab.vue

@@ -154,7 +154,7 @@ const renderedRoot = computed<TreeExplorerNode<ComfyModelDef>>(() => {
             if (onlyChild.data.file_name === '(No Content)') {
               return '0'
             } else if (onlyChild.data.file_name === 'Loading') {
-              return '?'
+              return ''
             }
           }
         }

src/types/treeExplorerTypes.ts

@@ -51,7 +51,7 @@ export interface RenderedTreeExplorerNode<T = any> extends TreeExplorerNode<T> {
   type: 'folder' | 'node'
   /** Total number of leaves in the subtree */
   totalLeaves: number
-  /** Text to display on the leaf-count badge */
+  /** Text to display on the leaf-count badge. Empty string means no badge. */
   badgeText?: string
 }