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

add logs to confirm that event info is being sent to bus

Browse Source
This commit is contained in:
Lincoln Stein
2023-09-16 22:38:37 -04:00
parent 539776a15a
commit e880f4bcfb
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
invokeai/app/services/events.py
View File

@ -7,6 +7,8 @@ from invokeai.app.services.model_manager_service import BaseModelType, ModelInfo
from invokeai.app.util.misc import get_timestamp
from invokeai.backend.model_manager.download import DownloadJobBase
from invokeai.backend.util.logging import InvokeAILogger
class EventServiceBase:
"""Basic event bus, to have an empty stand-in when not needed."""
@ -180,6 +182,9 @@ class EventServiceBase:
def emit_model_event(self, job: DownloadJobBase):
"""Emit event when the status of a download/install job changes."""
logger = InvokeAILogger.getLogger()
progress = 100 * (job.bytes / job.total_bytes) if job.total_bytes > 0 else 0
logger.info(f"Dispatch model_event for job {job.id}, status={job.status.value}, progress={progress:5.2f}%")
self.dispatch( # use dispatch() directly here because we are not a session event.
event_name="model_event", payload=dict(job=job)
)