Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: Restore Delete Model Functionality

Browse Source
This commit is contained in:
blessedcoolant
2023-07-12 16:39:07 +12:00
parent afb46564e8
commit 5a6ad99d4e
3 changed files with 27 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
invokeai/frontend/web/src/services/api/endpoints/models.ts
View File

@ -39,6 +39,11 @@ type UpdateMainModelQuery = {
body: MainModelConfig;
};
type DeleteMainModelQuery = {
base_model: BaseModelType;
model_name: string;
};
const mainModelsAdapter = createEntityAdapter<MainModelConfigEntity>({
sortComparer: (a, b) => a.name.localeCompare(b.name),
});
@ -121,6 +126,18 @@ export const modelsApi = api.injectEndpoints({
},
invalidatesTags: ['MainModel'],
}),
deleteMainModels: build.mutation<
EntityState<MainModelConfigEntity>,
DeleteMainModelQuery
>({
query: ({ base_model, model_name }) => {
return {
url: `models/${base_model}/main/${model_name}`,
method: 'DELETE',
};
},
invalidatesTags: ['MainModel'],
}),
getLoRAModels: build.query<EntityState<LoRAModelConfigEntity>, void>({
query: () => ({ url: 'models/', params: { model_type: 'lora' } }),
providesTags: (result, error, arg) => {
@ -265,4 +282,5 @@ export const {
useGetTextualInversionModelsQuery,
useGetVaeModelsQuery,
useUpdateMainModelsMutation,
useDeleteMainModelsMutation,
} = modelsApi;