alphard/ComfyUI-Lora-Manager
1
0
Fork 0
mirror of https://github.com/willmiao/ComfyUI-Lora-Manager.git synced 2026-03-21 21:22:11 -03:00
Code Issues Packages Projects Releases Wiki Activity

Enhance get_cached_data method: improve cache rebuilding logic and ensure cache is saved after initialization

Browse Source
This commit is contained in:
Will Miao
2025-05-29 08:50:17 +08:00
parent 651e14c8c3
commit afb012029f
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
py/services/model_scanner.py
View File

@@ -475,17 +475,23 @@ class ModelScanner:
# If force refresh is requested, initialize the cache directly # If force refresh is requested, initialize the cache directly
if force_refresh: if force_refresh:
# If rebuild_cache is True, try to reload from disk before reconciliation # If rebuild_cache is True, try to reload from disk before reconciliation
if rebuild_cache and self._cache is not None: if rebuild_cache:
logger.info(f"{self.model_type.capitalize()} Scanner: Attempting to rebuild cache from disk...") logger.info(f"{self.model_type.capitalize()} Scanner: Attempting to rebuild cache from disk...")
cache_loaded = await self._load_cache_from_disk() cache_loaded = await self._load_cache_from_disk()
if cache_loaded: if cache_loaded:
logger.info(f"{self.model_type.capitalize()} Scanner: Successfully reloaded cache from disk") logger.info(f"{self.model_type.capitalize()} Scanner: Successfully reloaded cache from disk")
else: else:
logger.info(f"{self.model_type.capitalize()} Scanner: Could not reload cache from disk, proceeding with normal refresh") logger.info(f"{self.model_type.capitalize()} Scanner: Could not reload cache from disk, proceeding with complete rebuild")
# If loading from disk failed, do a complete rebuild and save to disk
await self._initialize_cache()
await self._save_cache_to_disk()
return self._cache
if self._cache is None: if self._cache is None:
# For initial creation, do a full initialization # For initial creation, do a full initialization
await self._initialize_cache() await self._initialize_cache()
# Save the newly built cache
await self._save_cache_to_disk()
else: else:
# For subsequent refreshes, use fast reconciliation # For subsequent refreshes, use fast reconciliation
await self._reconcile_cache() await self._reconcile_cache()