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

fix(preview): hide license icons for models without CivitAI metadata

Browse Source
This commit is contained in:
Will Miao
2026-07-13 19:49:10 +08:00
parent 966024e534
commit feab01cd9c
1 changed files with 14 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
py/routes/handlers/model_handlers.py
View File

@@ -1313,9 +1313,20 @@ class ModelQueryHandler:
} }
if include_license_flags: if include_license_flags:
model_data = await self._service.get_model_info_by_name(model_name) model_data = await self._service.get_model_info_by_name(model_name)
license_flags = (model_data or {}).get("license_flags") # Only return license_flags when real CivitAI model license
if license_flags is not None: # data exists. This mirrors ModelModal's guard
response_payload["license_flags"] = int(license_flags) # (modelData?.civitai?.model) so the preview tooltip never
# shows misleading license icons for HF or other models
# without actual license metadata.
civitai_data = (model_data or {}).get("civitai") or {}
has_license_data = (
isinstance(civitai_data, dict)
and isinstance(civitai_data.get("model"), dict)
)
if has_license_data:
license_flags = (model_data or {}).get("license_flags")
if license_flags is not None:
response_payload["license_flags"] = int(license_flags)
# Include the user's license icon style preference so the # Include the user's license icon style preference so the
# ComfyUI tooltip can pick the right set without a separate # ComfyUI tooltip can pick the right set without a separate
# API call. # API call.