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

Add additional verbose logs for cache

Browse source
This commit is contained in:
Andrei Betlen 2023-06-14 21:46:48 -04:00
parent 4cefb70cd0
commit f27393ab7e
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
llama_cpp/server/app.py
View file

@ -119,8 +119,12 @@ def create_app(settings: Optional[Settings] = None):
) )
if settings.cache: if settings.cache:
if settings.cache_type == "disk": if settings.cache_type == "disk":
if settings.verbose:
print(f"Using disk cache with size {settings.cache_size}")
cache = llama_cpp.LlamaDiskCache(capacity_bytes=settings.cache_size) cache = llama_cpp.LlamaDiskCache(capacity_bytes=settings.cache_size)
else: else:
if settings.verbose:
print(f"Using ram cache with size {settings.cache_size}")
cache = llama_cpp.LlamaRAMCache(capacity_bytes=settings.cache_size) cache = llama_cpp.LlamaRAMCache(capacity_bytes=settings.cache_size)
cache = llama_cpp.LlamaCache(capacity_bytes=settings.cache_size) cache = llama_cpp.LlamaCache(capacity_bytes=settings.cache_size)