17 lines
455 B
Python
17 lines
455 B
Python
class AttentionBank:
|
|
def __init__(self, save_steps, block_map, inject_steps=None):
|
|
self._data = {
|
|
"save_steps": save_steps,
|
|
"block_map": block_map,
|
|
"inject_steps": inject_steps,
|
|
}
|
|
|
|
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)
|