Skip to content

Commit 5f12b91

Browse files
feat(mm): add cache_snapshot to model cache clear callback
1 parent d958d2e commit 5f12b91

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

invokeai/backend/model_manager/load/model_cache/model_cache.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,6 +84,7 @@ def __call__(
8484
models_cleared: int,
8585
bytes_requested: int,
8686
bytes_freed: int,
87+
cache_snapshot: dict[str, CacheEntrySnapshot],
8788
) -> None: ...
8889

8990

@@ -733,6 +734,7 @@ def make_room(self, bytes_needed: int) -> None:
733734
models_cleared=models_cleared,
734735
bytes_requested=bytes_needed,
735736
bytes_freed=ram_bytes_freed,
737+
cache_snapshot=self._get_cache_snapshot(),
736738
)
737739
gc.collect()
738740

0 commit comments

Comments
 (0)