Files
logtd-ComfyUI-Fluxtapoz/utils/attn_bank.py
T
2024-12-02 19:36:50 -07:00

17 lines
350 B
Python

class AttentionBank:
def __init__(self):
self._data = {}
def __getitem__(self, key):
return self._data[key]
def __setitem__(self, key, value):
self._data[key] = value
def get(self, key, default=None):
return self._data.get(key, default)
def clear(self):
self._data.clear()