Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

replace config.ram_cache_size with config.ram and similarly for vram

Browse Source
This commit is contained in:
Lincoln Stein
2023-10-03 15:36:23 -04:00
parent 062a6ed180
commit e3912e8826
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
invokeai/backend/model_manager/loader.py
View File

@ -158,13 +158,13 @@ class ModelLoad(ModelLoadBase):
self._logger.info(f"Using models database {models_file}")
self._logger.info(f"Rendering device = {device} ({device_name})")
self._logger.info(f"Maximum RAM cache size: {config.ram_cache_size}")
self._logger.info(f"Maximum VRAM cache size: {config.vram_cache_size}")
self._logger.info(f"Maximum RAM cache size: {config.ram}")
self._logger.info(f"Maximum VRAM cache size: {config.vram}")
self._logger.info(f"Precision: {precision}")
self._cache = ModelCache(
max_cache_size=config.ram_cache_size,
max_vram_cache_size=config.vram_cache_size,
max_cache_size=config.ram,
max_vram_cache_size=config.vram,
lazy_offloading=config.lazy_offload,
execution_device=device,
precision=dtype,