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: Add Manual Checkpoint / Safetensor Models

Browse Source
This commit is contained in:
blessedcoolant
2023-07-16 15:21:49 +12:00
parent 421fcb761b
commit d93d42af4a
6 changed files with 145 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -96,6 +96,9 @@ type AddMainModelResponse =
type SearchFolderResponse =
paths['/api/v1/models/search']['get']['responses']['200']['content']['application/json'];
type CheckpointConfigsResponse =
paths['/api/v1/models/ckpt_confs']['get']['responses']['200']['content']['application/json'];
type SearchFolderArg = operations['search_for_models']['parameters']['query'];
const mainModelsAdapter = createEntityAdapter<MainModelConfigEntity>({
@ -383,6 +386,13 @@ export const modelsApi = api.injectEndpoints({
};
},
}),
getCheckpointConfigs: build.query<CheckpointConfigsResponse, void>({
query: () => {
return {
url: `/models/ckpt_confs`,
};
},
}),
}),
});
@ -399,4 +409,5 @@ export const {
useConvertMainModelsMutation,
useMergeMainModelsMutation,
useGetModelsInFolderQuery,
useGetCheckpointConfigsQuery,
} = modelsApi;