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

Correct issues caused by merging main

Browse Source
This commit is contained in:
Brandon Rising
2023-07-26 12:25:46 -04:00
parent c16da75ac7
commit 861c0fe76b
13 changed files with 613 additions and 601 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -153,7 +153,15 @@ const createModelEntities = <T extends AnyModelConfigEntity>(
export const modelsApi = api.injectEndpoints({
endpoints: (build) => ({
getOnnxModels: build.query<EntityState<OnnxModelConfigEntity>, void>({
query: () => ({ url: 'models/', params: { model_type: 'onnx' } }),
query: (base_models) => {
const params = {
model_type: 'onnx',
base_models,
};
const query = queryString.stringify(params, { arrayFormat: 'none' });
return `models/?${query}`;
},
providesTags: (result, error, arg) => {
const tags: ApiFullTagDescription[] = [
{ id: 'OnnxModel', type: LIST_TAG },