diff --git a/core/src/types/model/modelSource.ts b/core/src/types/model/modelSource.ts index af1e75d72..ecf3e8aee 100644 --- a/core/src/types/model/modelSource.ts +++ b/core/src/types/model/modelSource.ts @@ -61,6 +61,7 @@ export interface ModelSibling { */ export interface ModelSource { id: string + author?: string metadata: Metadata models: ModelSibling[] type?: string diff --git a/extensions/inference-cortex-extension/bin/version.txt b/extensions/inference-cortex-extension/bin/version.txt index 428c6e50d..5bf81f993 100644 --- a/extensions/inference-cortex-extension/bin/version.txt +++ b/extensions/inference-cortex-extension/bin/version.txt @@ -1 +1 @@ -1.0.11-rc3 +1.0.11-rc4 diff --git a/extensions/model-extension/resources/default.json b/extensions/model-extension/resources/default.json index 4c33789a1..946f7cea2 100644 --- a/extensions/model-extension/resources/default.json +++ b/extensions/model-extension/resources/default.json @@ -1,6 +1,6 @@ [ { - "author": "cortexso", + "author": "CohereForAI", "id": "cortexso/command-r", "metadata": { "_id": "66751b98585f2bf57092b2ae", @@ -90,7 +90,7 @@ ] }, { - "author": "cortexso", + "author": "Google", "id": "cortexso/gemma2", "metadata": { "_id": "66b06c37491b555fefe0a0bf", @@ -264,7 +264,7 @@ ] }, { - "author": "cortexso", + "author": "CohereForAI", "id": "cortexso/aya", "metadata": { "_id": "66790e21db26e8589ccd3816", @@ -314,7 +314,7 @@ ] }, { - "author": "cortexso", + "author": "Qwen", "id": "cortexso/qwen2.5", "metadata": { "_id": "671d0d55748faf685e6450a3", @@ -740,7 +740,7 @@ ] }, { - "author": "cortexso", + "author": "meta-llama", "id": "cortexso/llama3.2", "metadata": { "_id": "66f63309ba963b1db95deaa4", @@ -888,7 +888,7 @@ ] }, { - "author": "cortexso", + "author": "DeepSeek-AI", "id": "cortexso/deepseek-r1", "metadata": { "_id": "67a0bcf13ac2dd6adf0bdfcf", @@ -1142,7 +1142,7 @@ ] }, { - "author": "cortexso", + "author": "meta-llama", "id": "cortexso/llama3.1", "metadata": { "_id": "66a76e01a1037fe261a5a472", diff --git a/web/screens/Hub/ModelList/ModelHeader/index.tsx b/web/screens/Hub/ModelList/ModelHeader/index.tsx index e02fc4fa8..5fe0993ff 100644 --- a/web/screens/Hub/ModelList/ModelHeader/index.tsx +++ b/web/screens/Hub/ModelList/ModelHeader/index.tsx @@ -34,10 +34,6 @@ import { getDownloadingModelAtom, } from '@/helpers/atoms/Model.atom' import { selectedSettingAtom } from '@/helpers/atoms/Setting.atom' -import { - nvidiaTotalVramAtom, - totalRamAtom, -} from '@/helpers/atoms/SystemBar.atom' type Props = { model: ModelSource @@ -50,14 +46,9 @@ const ModelItemHeader = ({ model, onSelectedModel }: Props) => { const downloadedModels = useAtomValue(downloadedModelsAtom) const setSelectedSetting = useSetAtom(selectedSettingAtom) const { requestCreateNewThread } = useCreateNewThread() - const totalRam = useAtomValue(totalRamAtom) - const nvidiaTotalVram = useAtomValue(nvidiaTotalVramAtom) const setMainViewState = useSetAtom(mainViewStateAtom) - // Default nvidia returns vram in MB, need to convert to bytes to match the unit of totalRamW - const ram = nvidiaTotalVram > 0 ? nvidiaTotalVram * 1024 * 1024 : totalRam - const serverEnabled = useAtomValue(serverEnabledAtom) const assistants = useAtomValue(assistantsAtom) diff --git a/web/screens/Hub/ModelList/ModelItem/index.tsx b/web/screens/Hub/ModelList/ModelItem/index.tsx index 42e1b74f2..e782e6f70 100644 --- a/web/screens/Hub/ModelList/ModelItem/index.tsx +++ b/web/screens/Hub/ModelList/ModelItem/index.tsx @@ -41,10 +41,10 @@ const ModelItem: React.FC = ({ model, onSelectedModel }) => {
- {model.metadata?.author && ( + {(model?.author ?? model?.metadata?.author) && (

{model.id?.includes('huggingface.co') && ( <> @@ -57,7 +57,7 @@ const ModelItem: React.FC = ({ model, onSelectedModel }) => { />{' '} )}{' '} - {model.metadata?.author} + {model?.author ?? model?.metadata?.author}

)} {model.models?.length > 0 && ( diff --git a/web/screens/Hub/ModelPage/index.tsx b/web/screens/Hub/ModelPage/index.tsx index dfa89c7f1..b52ae72a9 100644 --- a/web/screens/Hub/ModelPage/index.tsx +++ b/web/screens/Hub/ModelPage/index.tsx @@ -102,10 +102,10 @@ const ModelPage = ({ model, onGoBack }: Props) => {
- {model.metadata?.author && ( + {(model?.author ?? model?.metadata?.author) && (

{model.id?.includes('huggingface.co') && ( <> @@ -118,7 +118,7 @@ const ModelPage = ({ model, onGoBack }: Props) => { />{' '} )} - {model.metadata?.author} + {model?.author ?? model?.metadata?.author}

)} {model.models?.length > 0 && (