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

tidy(nodes): do not store unnecessarily store invoker

Browse Source
This commit is contained in:
psychedelicious 2024-02-10 09:41:23 +11:00
parent c58f8c3269
commit ff249a2315
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/app/services/object_serializer/object_serializer_disk.py
View File

@ -37,13 +37,11 @@ class ObjectSerializerDisk(ObjectSerializerBase[T]):
self.__obj_class_name: Optional[str] = None self.__obj_class_name: Optional[str] = None
def start(self, invoker: "Invoker") -> None: def start(self, invoker: "Invoker") -> None:
self._invoker = invoker
if self._delete_on_startup: if self._delete_on_startup:
delete_all_result = self._delete_all() delete_all_result = self._delete_all()
if delete_all_result.deleted_count > 0: if delete_all_result.deleted_count > 0:
freed_space_in_mb = round(delete_all_result.freed_space_bytes / 1024 / 1024, 2) freed_space_in_mb = round(delete_all_result.freed_space_bytes / 1024 / 1024, 2)
self._invoker.services.logger.info( invoker.services.logger.info(
f"Deleted {delete_all_result.deleted_count} {self._obj_class_name} files (freed {freed_space_in_mb}MB)" f"Deleted {delete_all_result.deleted_count} {self._obj_class_name} files (freed {freed_space_in_mb}MB)"
) )