baalajimaestro/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Cache should raise KeyError when key is missing

Browse source
This commit is contained in:
Andrei Betlen 2023-05-05 12:21:49 -04:00
parent 24fc38754b
commit 5be0efa5f8
1 changed files with 2 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
llama_cpp/llama.py
View file

@ -33,12 +33,10 @@ class LlamaCache:
return k return k
return None return None
def __getitem__( def __getitem__(self, key: Sequence[llama_cpp.llama_token]) -> "LlamaState":
self, key: Sequence[llama_cpp.llama_token]
) -> Optional["LlamaState"]:
_key = self._find_key(tuple(key)) _key = self._find_key(tuple(key))
if _key is None: if _key is None:
return None raise KeyError(f"Key not found: {key}")
return self.cache_state[_key] return self.cache_state[_key]
def __contains__(self, key: Sequence[llama_cpp.llama_token]) -> bool: def __contains__(self, key: Sequence[llama_cpp.llama_token]) -> bool: