26 lines
905 B
TypeScript
26 lines
905 B
TypeScript
import { ModelVersion } from '@/_models/ModelVersion'
|
|
import { useState } from 'react'
|
|
import { useAtomValue } from 'jotai'
|
|
import { totalRamAtom } from '@/_helpers/atoms/SystemBar.atom'
|
|
|
|
export default function useGetMostSuitableModelVersion() {
|
|
const [suitableModel, setSuitableModel] = useState<ModelVersion | undefined>()
|
|
const totalRam = useAtomValue(totalRamAtom)
|
|
|
|
const getMostSuitableModelVersion = async (modelVersions: ModelVersion[]) => {
|
|
// find the model version with the highest required RAM that is still below the user's RAM by 80%
|
|
const modelVersion = modelVersions.reduce((prev, current) => {
|
|
if (current.maxRamRequired > prev.maxRamRequired) {
|
|
if (current.maxRamRequired < totalRam * 0.8) {
|
|
return current
|
|
}
|
|
}
|
|
return prev
|
|
})
|
|
|
|
setSuitableModel(modelVersion)
|
|
}
|
|
|
|
return { suitableModel, getMostSuitableModelVersion }
|
|
}
|