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

feat(nodes): create helper function to generate the item ID

Browse Source
This commit is contained in:
psychedelicious 2024-02-07 19:50:30 +11:00
parent fe0391c86b
commit 7fe5283e74
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
invokeai/app/services/item_storage/item_storage_ephemeral_disk.py
View File

@ -37,7 +37,7 @@ class ItemStorageEphemeralDisk(ItemStorageABC[T]):
def set(self, item: T) -> str: def set(self, item: T) -> str:
self._output_folder.mkdir(parents=True, exist_ok=True) self._output_folder.mkdir(parents=True, exist_ok=True)
item_id = f"{self._item_class_name}_{uuid_string()}" item_id = self._new_item_id()
file_path = self._get_path(item_id) file_path = self._get_path(item_id)
torch.save(item, file_path) # pyright: ignore [reportUnknownMemberType] torch.save(item, file_path) # pyright: ignore [reportUnknownMemberType]
return item_id return item_id
@ -49,6 +49,9 @@ class ItemStorageEphemeralDisk(ItemStorageABC[T]):
def _get_path(self, item_id: str) -> Path: def _get_path(self, item_id: str) -> Path:
return self._output_folder / item_id return self._output_folder / item_id
def _new_item_id(self) -> str:
return f"{self._item_class_name}_{uuid_string()}"
def _delete_all_items(self) -> None: def _delete_all_items(self) -> None:
""" """
Deletes all pickled items from disk. Deletes all pickled items from disk.