github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-02-17 09:01:48 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat(ui): add LoRA ui & update graphs

Browse Source
This commit is contained in:
psychedelicious
2023-07-04 21:12:52 +10:00
parent d537b9f0cb
commit db8862d860
26 changed files with 630 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,6 +1,6 @@
import { ModelsList } from 'services/api/types';
import { EntityState, createEntityAdapter } from '@reduxjs/toolkit';
import { keyBy } from 'lodash-es';
import { ModelsList } from 'services/api/types';
import { ApiFullTagDescription, LIST_TAG, api } from '..';
import { paths } from '../schema';
@@ -24,11 +24,9 @@ export const modelsApi = api.injectEndpoints({
listModels: build.query<EntityState<ModelConfig>, ListModelsArg>({
query: (arg) => ({ url: 'models/', params: arg }),
providesTags: (result, error, arg) => {
// any list of boards
const tags: ApiFullTagDescription[] = [{ id: 'Model', type: LIST_TAG }];
if (result) {
// and individual tags for each board
tags.push(
...result.ids.map((id) => ({
type: 'Model' as const,

3
invokeai/frontend/web/src/services/api/types.d.ts vendored
View File

@@ -35,7 +35,9 @@ export type ModelType = S<'ModelType'>;
export type BaseModelType = S<'BaseModelType'>;
export type MainModelField = S<'MainModelField'>;
export type VAEModelField = S<'VAEModelField'>;
export type LoRAModelField = S<'LoRAModelField'>;
export type ModelsList = S<'ModelsList'>;
export type LoRAModelConfig = S<'LoRAModelConfig'>;
// Graphs
export type Graph = S<'Graph'>;
@@ -60,6 +62,7 @@ export type ImageToLatentsInvocation = N<'ImageToLatentsInvocation'>;
export type LatentsToImageInvocation = N<'LatentsToImageInvocation'>;
export type ImageCollectionInvocation = N<'ImageCollectionInvocation'>;
export type MainModelLoaderInvocation = N<'MainModelLoaderInvocation'>;
export type LoraLoaderInvocation = N<'LoraLoaderInvocation'>;
// ControlNet Nodes
export type ControlNetInvocation = N<'ControlNetInvocation'>;