26 lines
917 B
TypeScript
26 lines
917 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 };
|
|
}
|