Nicholai/jan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added calculation active model size into available RAM

Browse Source
This commit is contained in:
Faisal Amir 2024-01-10 17:04:32 +07:00
parent aca6fee091
commit 487f4f7891
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
web/screens/ExploreModels/ExploreModelItemHeader/index.tsx
View File

@ -14,6 +14,7 @@ import ModalCancelDownload from '@/containers/ModalCancelDownload'
import { MainViewState } from '@/constants/screens' import { MainViewState } from '@/constants/screens'
import { useActiveModel } from '@/hooks/useActiveModel'
import { useCreateNewThread } from '@/hooks/useCreateNewThread' import { useCreateNewThread } from '@/hooks/useCreateNewThread'
import useDownloadModel from '@/hooks/useDownloadModel' import useDownloadModel from '@/hooks/useDownloadModel'
import { useDownloadState } from '@/hooks/useDownloadState' import { useDownloadState } from '@/hooks/useDownloadState'
@ -83,9 +84,11 @@ const ExploreModelItemHeader: React.FC<Props> = ({ model, onClick, open }) => {
downloadButton = <ModalCancelDownload model={model} /> downloadButton = <ModalCancelDownload model={model} />
} }
const { activeModel } = useActiveModel()
const getLabel = (size: number) => { const getLabel = (size: number) => {
const minimumRamModel = size * 1.25 const minimumRamModel = size * 1.25
const availableRam = totalRam - usedRam const availableRam = totalRam - usedRam + (activeModel?.metadata.size ?? 0)
if (minimumRamModel > totalRam) { if (minimumRamModel > totalRam) {
return ( return (