chore: update progress bar variant model

This commit is contained in:
Faisal Amir 2025-08-14 10:22:57 +07:00
parent da08becd93
commit dd5b9e23f5

View File

@ -139,7 +139,7 @@ function Hub() {
filtered = filtered?.filter((model) => filtered = filtered?.filter((model) =>
model.quants.some((variant) => model.quants.some((variant) =>
llamaProvider?.models.some( llamaProvider?.models.some(
(m: { id: string }) => m.id === variant.model_id (m: { id: string }) => m.id === sanitizeModelId(variant.model_id)
) )
) )
) )
@ -677,8 +677,8 @@ function Hub() {
{filteredModels[virtualItem.index].quants.map( {filteredModels[virtualItem.index].quants.map(
(variant) => ( (variant) => (
<CardItem <CardItem
key={variant.model_id} key={sanitizeModelId(variant.model_id)}
title={variant.model_id} title={sanitizeModelId(variant.model_id)}
actions={ actions={
<div className="flex items-center gap-2"> <div className="flex items-center gap-2">
<p className="text-main-view-fg/70 font-medium text-xs"> <p className="text-main-view-fg/70 font-medium text-xs">
@ -687,19 +687,32 @@ function Hub() {
{(() => { {(() => {
const isDownloading = const isDownloading =
localDownloadingModels.has( localDownloadingModels.has(
sanitizeModelId(
variant.model_id variant.model_id
)
) || ) ||
downloadProcesses.some( downloadProcesses.some(
(e) => e.id === variant.model_id (e) =>
e.id ===
sanitizeModelId(
variant.model_id
)
) )
const downloadProgress = const downloadProgress =
downloadProcesses.find( downloadProcesses.find(
(e) => e.id === variant.model_id (e) =>
e.id ===
sanitizeModelId(
variant.model_id
)
)?.progress || 0 )?.progress || 0
const isDownloaded = const isDownloaded =
llamaProvider?.models.some( llamaProvider?.models.some(
(m: { id: string }) => (m: { id: string }) =>
m.id === variant.model_id m.id ===
sanitizeModelId(
variant.model_id
)
) )
if (isDownloading) { if (isDownloading) {
@ -733,8 +746,10 @@ function Hub() {
size="sm" size="sm"
onClick={() => onClick={() =>
handleUseModel( handleUseModel(
sanitizeModelId(
variant.model_id variant.model_id
) )
)
} }
> >
{t('hub:use')} {t('hub:use')}
@ -749,10 +764,18 @@ function Hub() {
title={t('hub:downloadModel')} title={t('hub:downloadModel')}
onClick={() => { onClick={() => {
addLocalDownloadingModel( addLocalDownloadingModel(
sanitizeModelId(variant.model_id) sanitizeModelId(
sanitizeModelId(
variant.model_id
)
)
) )
pullModel( pullModel(
sanitizeModelId(variant.model_id), sanitizeModelId(
sanitizeModelId(
variant.model_id
)
),
variant.path variant.path
) )
}} }}