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

initiate TI model query when socket connects so user doesnt have to wait when opening prompt trigger phrases

Browse Source
This commit is contained in:
Mary Hipp 2024-03-13 13:19:19 -04:00 committed by psychedelicious
parent 1d2743af1b
commit cd55c23713
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/socketio/socketConnected.ts
View File

@ -4,6 +4,7 @@ import { $baseUrl } from 'app/store/nanostores/baseUrl';
import { isEqual } from 'lodash-es'; import { isEqual } from 'lodash-es';
import { atom } from 'nanostores'; import { atom } from 'nanostores';
import { api } from 'services/api'; import { api } from 'services/api';
import { modelsApi } from 'services/api/endpoints/models';
import { queueApi, selectQueueStatus } from 'services/api/endpoints/queue'; import { queueApi, selectQueueStatus } from 'services/api/endpoints/queue';
import { socketConnected } from 'services/events/actions'; import { socketConnected } from 'services/events/actions';
@ -17,6 +18,9 @@ export const addSocketConnectedEventListener = (startAppListening: AppStartListe
effect: async (action, { dispatch, getState, cancelActiveListeners, delay }) => { effect: async (action, { dispatch, getState, cancelActiveListeners, delay }) => {
log.debug('Connected'); log.debug('Connected');
// query TIs so they are ready if user opens trigger phrases
dispatch(modelsApi.endpoints.getTextualInversionModels.initiate());
/** /**
* The rest of this listener has recovery logic for when the socket disconnects and reconnects. * The rest of this listener has recovery logic for when the socket disconnects and reconnects.
* *