github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

Update invokeai/app/api/routers/model_manager.py

Browse Source 
Co-authored-by: Ryan Dick <ryanjdick3@gmail.com>
This commit is contained in:
Lincoln Stein
2024-06-21 15:15:31 -04:00
committed by GitHub
parent 4c5bad6352
commit 787671c2c2
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
invokeai/app/api/routers/model_manager.py
View File

@@ -825,7 +825,12 @@ async def get_starter_models() -> list[StarterModel]:
async def get_cache_size(cache_type: CacheType = Query(description="The cache type", default=CacheType.RAM)) -> float:
"""Return the current RAM or VRAM cache size setting (in GB)."""
cache = ApiDependencies.invoker.services.model_manager.load.ram_cache
return cache.max_cache_size if cache_type == CacheType.RAM else cache.max_vram_cache_size
if cache_type == CacheType.RAM:
return cache.max_cache_size
elif cache_type == CacheType.VRAM:
return cache.max_vram_cache_size
else:
raise ValueError(f"Unexpected {cache_type=}.")
@model_manager_router.put(