diff --git a/tools/ui/src/lib/components/app/models/ModelAvatar.svelte b/tools/ui/src/lib/components/app/models/ModelAvatar.svelte index ddb5f2f96e..7f9ca08f73 100644 --- a/tools/ui/src/lib/components/app/models/ModelAvatar.svelte +++ b/tools/ui/src/lib/components/app/models/ModelAvatar.svelte @@ -18,6 +18,9 @@ * corner badge. Resolving the base org costs one Hugging Face request per repo, so * it waits until the row is near the viewport. */ showBaseModelAvatar?: boolean; + /** Show the repo's own org as the main image, skipping the base model. + * Used inside a heading that already carries the base org. */ + showRepoOrgAvatar?: boolean; /** Keep the quantizer badge off, e.g. when a row stands for a whole family. */ showQuantBadge?: boolean; size?: string; @@ -30,6 +33,7 @@ quantSize = 'h-3 w-3', showBaseModelAvatar = false, showQuantBadge = true, + showRepoOrgAvatar = false, size = 'size-5' }: Props = $props(); @@ -113,8 +117,8 @@ - + - + (highlightedId = option.id)} onSelect={ms.handleSelect} {option} - showBaseModelAvatar + showBaseModelAvatar={!settingsStore.config.groupModelsByFamily} + showRepoOrgAvatar={settingsStore.config.groupModelsByFamily} /> {/snippet} diff --git a/tools/ui/src/lib/components/app/models/ModelsSelector/ModelsSelectorList.svelte b/tools/ui/src/lib/components/app/models/ModelsSelector/ModelsSelectorList.svelte index 51ce158655..df4b0edf11 100644 --- a/tools/ui/src/lib/components/app/models/ModelsSelector/ModelsSelectorList.svelte +++ b/tools/ui/src/lib/components/app/models/ModelsSelector/ModelsSelectorList.svelte @@ -156,7 +156,8 @@ onMouseEnter={() => {}} {onSelect} {option} - showBaseModelAvatar + showBaseModelAvatar={!settingsStore.config.groupModelsByFamily} + showRepoOrgAvatar={settingsStore.config.groupModelsByFamily} /> {/snippet} diff --git a/tools/ui/src/lib/components/app/models/ModelsSelector/ModelsSelectorOption.svelte b/tools/ui/src/lib/components/app/models/ModelsSelector/ModelsSelectorOption.svelte index 19bdd0a23f..cfd88b58d4 100644 --- a/tools/ui/src/lib/components/app/models/ModelsSelector/ModelsSelectorOption.svelte +++ b/tools/ui/src/lib/components/app/models/ModelsSelector/ModelsSelectorOption.svelte @@ -19,6 +19,8 @@ onKeyDown: (e: KeyboardEvent) => void; /** Show the base model's org as the main avatar and the repo (quant) org as the corner badge; resolves the base org lazily via HF. */ showBaseModelAvatar?: boolean; + /** Show the repo's own org instead of the base model's. */ + showRepoOrgAvatar?: boolean; } let { @@ -30,7 +32,8 @@ onMouseEnter, onSelect, option, - showBaseModelAvatar = false + showBaseModelAvatar = false, + showRepoOrgAvatar = false }: Props = $props(); // row actions follow the backend that serves the row, not the selected one @@ -86,7 +89,7 @@ tabindex="0" title={loadTitle} > - +