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 error_reason to ModelInstallJob

Browse Source
This commit is contained in:
maryhipp 2024-02-23 10:40:51 -05:00 committed by psychedelicious
parent 625c86ba9a
commit 26a209a00d
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/app/services/model_install/model_install_base.py
View File

@ -156,6 +156,7 @@ class ModelInstallJob(BaseModel):
id: int = Field(description="Unique ID for this job") id: int = Field(description="Unique ID for this job")
status: InstallStatus = Field(default=InstallStatus.WAITING, description="Current status of install process") status: InstallStatus = Field(default=InstallStatus.WAITING, description="Current status of install process")
error_reason: Optional[str] = Field(default=None, description="Information about why the job failed")
config_in: Dict[str, Any] = Field( config_in: Dict[str, Any] = Field(
default_factory=dict, description="Configuration information (e.g. 'description') to apply to model." default_factory=dict, description="Configuration information (e.g. 'description') to apply to model."
) )
@ -193,6 +194,7 @@ class ModelInstallJob(BaseModel):
self.error = str(e) self.error = str(e)
self.error_traceback = self._format_error(e) self.error_traceback = self._format_error(e)
self.status = InstallStatus.ERROR self.status = InstallStatus.ERROR
self.error_reason = self._exception.__class__.__name__ if self._exception else None
def cancel(self) -> None: def cancel(self) -> None:
"""Call to cancel the job.""" """Call to cancel the job."""