added hf models import tab and route for getting available hf models

This commit is contained in:
Jennifer Player
2024-03-07 16:57:28 -05:00
committed by psychedelicious
parent efea1a8a7d
commit f7cd3cf1f4
10 changed files with 308 additions and 3 deletions

View File

@@ -258,6 +258,13 @@ export const modelsApi = api.injectEndpoints({
};
},
}),
getHuggingFaceModels: build.query<string[], string>({
query: (hugging_face_repo) => {
return {
url: buildModelsUrl(`hugging_face?hugging_face_repo=${hugging_face_repo}`),
};
},
}),
listModelInstalls: build.query<ListModelInstallsResponse, void>({
query: () => {
return {
@@ -381,6 +388,7 @@ export const {
useConvertModelMutation,
useSyncModelsMutation,
useLazyScanFolderQuery,
useLazyGetHuggingFaceModelsQuery,
useListModelInstallsQuery,
useCancelModelInstallMutation,
usePruneCompletedModelInstallsMutation,