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

fix(ui): use model names in badges

Browse Source
This commit is contained in:
psychedelicious
2024-02-21 19:42:36 +11:00
parent 34f3a39cc9
commit 9d9b417432
8 changed files with 85 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -236,6 +236,18 @@ export const modelsApi = api.injectEndpoints({
},
invalidatesTags: ['Model'],
}),
getModelConfig: build.query<AnyModelConfig, string>({
query: (key) => buildModelsUrl(`i/${key}`),
providesTags: (result) => {
const tags: ApiTagDescription[] = ['Model'];
if (result) {
tags.push({ type: 'ModelConfig', id: result.key });
}
return tags;
},
}),
syncModels: build.mutation<SyncModelsResponse, void>({
query: () => {
return {
@ -313,6 +325,7 @@ export const modelsApi = api.injectEndpoints({
});
export const {
useGetModelConfigQuery,
useGetMainModelsQuery,
useGetControlNetModelsQuery,
useGetIPAdapterModelsQuery,