diff --git a/web/screens/HubScreen2/components/ModelSearchBar.tsx b/web/screens/HubScreen2/components/ModelSearchBar.tsx index 6b7ce3f5a..6ca3da884 100644 --- a/web/screens/HubScreen2/components/ModelSearchBar.tsx +++ b/web/screens/HubScreen2/components/ModelSearchBar.tsx @@ -10,8 +10,6 @@ import { toaster } from '@/containers/Toast' import { useGetHFRepoData } from '@/hooks/useGetHFRepoData' -// import { setImportModelStageAtom } from '@/hooks/useImportModel' - import { MainViewState, mainViewStateAtom } from '@/helpers/atoms/App.atom' import { importHuggingFaceModelStageAtom, @@ -28,7 +26,6 @@ const ModelSearchBar: React.FC = ({ onSearchChanged }) => { const { getHfRepoData } = useGetHFRepoData() const setMainViewState = useSetAtom(mainViewStateAtom) const setSelectedSetting = useSetAtom(selectedSettingAtom) - // const setImportModelStage = useSetAtom(setImportModelStageAtom) const setImportingHuggingFaceRepoData = useSetAtom( importingHuggingFaceRepoDataAtom @@ -37,10 +34,6 @@ const ModelSearchBar: React.FC = ({ onSearchChanged }) => { importHuggingFaceModelStageAtom ) - // const onImportModelClick = useCallback(() => { - // setImportModelStage('SELECTING_MODEL') - // }, [setImportModelStage]) - const debounced = useDebouncedCallback(async (searchText: string) => { if (searchText.indexOf('/') === -1) { // If we don't find / in the text, perform a local search @@ -97,15 +90,6 @@ const ModelSearchBar: React.FC = ({ onSearchChanged }) => { My models - {/* Temporary hidden button import model */} - {/* */} ) } diff --git a/web/utils/huggingface.ts b/web/utils/huggingface.ts index 803fcfed2..1bd71dff4 100644 --- a/web/utils/huggingface.ts +++ b/web/utils/huggingface.ts @@ -305,7 +305,11 @@ export const fetchHuggingFaceRepoData = async ( AllQuantizations.forEach((quantization) => { data.siblings.forEach((sibling) => { - if (!sibling.quantization && sibling.rfilename.includes(quantization)) { + if ( + !sibling.quantization && + (sibling.rfilename.includes(quantization) || + sibling.rfilename.includes(quantization.toLowerCase())) + ) { sibling.quantization = quantization } })