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

fix(routes): add support for metadata archive settings in model processing

Browse Source
This commit is contained in:
Will Miao
2025-09-11 09:31:58 +08:00
parent a1a55a1002
commit 1edf7126bb
2 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
py/routes/base_model_routes.py
View File

@@ -612,8 +612,9 @@ class BaseModelRoutes(ABC):
needs_resort = False needs_resort = False
# Prepare models to process, only those without CivitAI data or missing tags, description, or creator # Prepare models to process, only those without CivitAI data or missing tags, description, or creator
enable_metadata_archive_db = settings.get('enable_metadata_archive_db', False)
to_process = [ to_process = [
model for model in cache.raw_data model for model in cache.raw_data
if ( if (
model.get('sha256') model.get('sha256')
and ( and (
@@ -623,6 +624,10 @@ class BaseModelRoutes(ABC):
or not model.get('modelDescription') or not model.get('modelDescription')
or not (model.get('civitai') and model['civitai'].get('creator')) or not (model.get('civitai') and model['civitai'].get('creator'))
) )
and (
(enable_metadata_archive_db)
or (not enable_metadata_archive_db and model.get('from_civitai') is True)
)
) )
] ]
total_to_process = len(to_process) total_to_process = len(to_process)

3
py/utils/routes_common.py
View File

@@ -13,6 +13,7 @@ from ..utils.exif_utils import ExifUtils
from ..utils.metadata_manager import MetadataManager from ..utils.metadata_manager import MetadataManager
from ..services.websocket_manager import ws_manager from ..services.websocket_manager import ws_manager
from ..services.metadata_service import get_default_metadata_provider, get_metadata_provider from ..services.metadata_service import get_default_metadata_provider, get_metadata_provider
from ..services.settings_manager import settings
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@@ -207,6 +208,8 @@ class ModelRouteUtils:
local_metadata = await ModelRouteUtils.load_local_metadata(metadata_path) local_metadata = await ModelRouteUtils.load_local_metadata(metadata_path)
if model_data.get('from_civitai') is False: if model_data.get('from_civitai') is False:
if not settings.get('enable_metadata_archive_db', False):
return False
# Likely deleted from CivitAI, use archive_db if available # Likely deleted from CivitAI, use archive_db if available
metadata_provider = await get_metadata_provider('sqlite') metadata_provider = await get_metadata_provider('sqlite')
else: else: