diff --git a/invokeai/app/api/sockets.py b/invokeai/app/api/sockets.py index 8fa34da078..bfe38a3e13 100644 --- a/invokeai/app/api/sockets.py +++ b/invokeai/app/api/sockets.py @@ -12,6 +12,11 @@ from invokeai.app.services.events.events_common import ( BulkDownloadErrorEvent, BulkDownloadEventBase, BulkDownloadStartedEvent, + DownloadCancelledEvent, + DownloadCompleteEvent, + DownloadErrorEvent, + DownloadProgressEvent, + DownloadStartedEvent, FastAPIEvent, InvocationCompleteEvent, InvocationDenoiseProgressEvent, @@ -21,6 +26,7 @@ from invokeai.app.services.events.events_common import ( ModelInstallCancelledEvent, ModelInstallCompleteEvent, ModelInstallDownloadProgressEvent, + ModelInstallDownloadsCompleteEvent, ModelInstallErrorEvent, ModelInstallStartedEvent, ModelLoadCompleteEvent, @@ -78,9 +84,15 @@ class SocketIO: register_events( { + DownloadCancelledEvent, + DownloadCompleteEvent, + DownloadErrorEvent, + DownloadProgressEvent, + DownloadStartedEvent, ModelLoadStartedEvent, ModelLoadCompleteEvent, ModelInstallDownloadProgressEvent, + ModelInstallDownloadsCompleteEvent, ModelInstallStartedEvent, ModelInstallCompleteEvent, ModelInstallCancelledEvent,