59 lines
1.4 KiB
TypeScript
59 lines
1.4 KiB
TypeScript
import React from 'react'
|
|
|
|
import { useAtomValue } from 'jotai'
|
|
|
|
import { useActiveModel } from '@/hooks/useActiveModel'
|
|
|
|
import { useSettings } from '@/hooks/useSettings'
|
|
|
|
import NotEnoughMemoryLabel from './NotEnoughMemoryLabel'
|
|
|
|
import RecommendedLabel from './RecommendedLabel'
|
|
|
|
import SlowOnYourDeviceLabel from './SlowOnYourDeviceLabel'
|
|
|
|
import {
|
|
availableVramAtom,
|
|
totalRamAtom,
|
|
usedRamAtom,
|
|
} from '@/helpers/atoms/SystemBar.atom'
|
|
|
|
type Props = {
|
|
size: number
|
|
}
|
|
|
|
const ModelLabel: React.FC<Props> = ({ size }) => {
|
|
const { activeModel } = useActiveModel()
|
|
const totalRam = useAtomValue(totalRamAtom)
|
|
const usedRam = useAtomValue(usedRamAtom)
|
|
const availableVram = useAtomValue(availableVramAtom)
|
|
const { settings } = useSettings()
|
|
|
|
const getLabel = (size: number) => {
|
|
const minimumRamModel = size * 1.25
|
|
const availableRam =
|
|
settings?.run_mode === 'gpu'
|
|
? availableVram * 1000000 // MB to bytes
|
|
: totalRam - usedRam + (activeModel?.metadata.size ?? 0)
|
|
if (minimumRamModel > totalRam) {
|
|
return (
|
|
<NotEnoughMemoryLabel
|
|
unit={settings?.run_mode === 'gpu' ? 'VRAM' : 'RAM'}
|
|
/>
|
|
)
|
|
}
|
|
if (minimumRamModel < availableRam) {
|
|
return <RecommendedLabel />
|
|
}
|
|
if (minimumRamModel < totalRam && minimumRamModel > availableRam) {
|
|
return <SlowOnYourDeviceLabel />
|
|
}
|
|
|
|
return null
|
|
}
|
|
|
|
return getLabel(size)
|
|
}
|
|
|
|
export default React.memo(ModelLabel)
|